diff --git a/diffimg/test.py b/diffimg/test.py
index eccd3c9..c9f3814 100755
--- a/diffimg/test.py
+++ b/diffimg/test.py
@@ -5,7 +5,7 @@
 
 import PIL
 
-from diff import diff
+from diffimg.diff import diff
 
 IMAGES_DIR = "images"