diff --git a/tests.py b/tests.py
index c5ddfe3..d89cec3 100644
--- a/tests.py
+++ b/tests.py
@@ -1,6 +1,6 @@
 import random
-from cellar_extractor import *
-from cellar_extractor.operative_extractions import Analyzer, Writing
+from cellar import *
+
 
 def cellar_csv_n():
     get_cellar(save_file='n', file_format='csv', sd='2022-01-01', max_ecli=100)