diff --git a/test/test.py b/test/test.py
index b49f018..81db814 100644
--- a/test/test.py
+++ b/test/test.py
@@ -1,11 +1,10 @@
 import os
 import pytest
-from backupill import Pill
+from backupill import *
 
 
 this_dir, this_filename = os.path.split(__file__)
 
 
 def test_generate_backup():
-    pill = Pill(asc_file=os.path.join(this_dir, "test.asc"))
-    pill.generate_backup()
+    generate_backup(asc_file=os.path.join(this_dir, "test.asc"))