diff --git a/test_zippy_detect.py b/test_zippy_detect.py
index 1ecd5b0..2274314 100644
--- a/test_zippy_detect.py
+++ b/test_zippy_detect.py
@@ -2,8 +2,8 @@
 
 import pytest, os, jsonlines, csv
 from warnings import warn
-from zippy import Zippy, EnsembledZippy, PRELUDE_STR, LzmaLlmDetector, BrotliLlmDetector, ZlibLlmDetector, CompressionEngine
-import zippy
+from zippy.zippy import Zippy, EnsembledZippy, PRELUDE_STR, LzmaLlmDetector, BrotliLlmDetector, ZlibLlmDetector, CompressionEngine
+import zippy.zippy
 
 AI_SAMPLE_DIR = 'samples/llm-generated/'
 HUMAN_SAMPLE_DIR = 'samples/human-generated/'