From 538afb14e968b0036104ead0c7e8a63ecc8d5f8f Mon Sep 17 00:00:00 2001
From: akamhy <64683866+akamhy@users.noreply.github.com>
Date: Thu, 7 May 2020 15:06:52 +0530
Subject: [PATCH] Update test_1.py

---
 tests/test_1.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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__":