diff --git a/tests/test_1.py b/tests/test_1.py
index 09ab492..51a815d 100644
--- a/tests/test_1.py
+++ b/tests/test_1.py
@@ -73,10 +73,10 @@ def test_get():
 
 def test_total_archives():
 
-    count1 = total_archives("https://en.wikipedia.org/wiki/Python (programming language)", UA=user_agent)
+    count1 = waybackpy.total_archives("https://en.wikipedia.org/wiki/Python (programming language)", UA=user_agent)
     assert count1 > 2000
 
-    count2 = total_archives("https://gaha.e4i3n.m5iai3kip6ied.cima/gahh2718gs/ahkst63t7gad8", UA=user_agent)
+    count2 = waybackpy.total_archives("https://gaha.e4i3n.m5iai3kip6ied.cima/gahh2718gs/ahkst63t7gad8", UA=user_agent)
     assert count2 == 0
 
 if __name__ == "__main__":