diff --git a/tests.py b/tests.py
index d89cec3..cea29f8 100644
--- a/tests.py
+++ b/tests.py
@@ -1,5 +1,5 @@
 import random
-from cellar import *
+from cellar_extractor import *
 
 
 def cellar_csv_n():