diff --git a/tests/test_utils.py b/tests/test_utils.py
new file mode 100644
index 0000000..70259d9
--- /dev/null
+++ b/tests/test_utils.py
@@ -0,0 +1,13 @@
+from waybackpy.utils import latest_version, DEFAULT_USER_AGENT
+from waybackpy.__version__ import __version__
+
+
+def test_default_user_agent():
+    assert (
+        DEFAULT_USER_AGENT
+        == "waybackpy %s - https://github.com/akamhy/waybackpy" % __version__
+    )
+
+
+def test_latest_version():
+    assert __version__ == latest_version(package_name="waybackpy")
diff --git a/waybackpy/utils.py b/waybackpy/utils.py
index 594f608..26cb7cd 100644
--- a/waybackpy/utils.py
+++ b/waybackpy/utils.py
@@ -4,8 +4,9 @@ from .__version__ import __version__
 DEFAULT_USER_AGENT = "waybackpy %s - https://github.com/akamhy/waybackpy" % __version__
 
 
-def latest_version(package_name, headers):
+def latest_version(package_name, user_agent=DEFAULT_USER_AGENT):
     request_url = "https://pypi.org/pypi/" + package_name + "/json"
+    headers = {"User-Agent": user_agent}
     response = requests.get(request_url, headers=headers)
     data = response.json()
     return data["info"]["version"]