diff --git a/tests.py b/tests.py
index 2b424fc..f659fec 100644
--- a/tests.py
+++ b/tests.py
@@ -13,7 +13,7 @@
     'haunted cabinet 2.jpg',
     'spooky.jpg',
     '23-11-332.jpg',
-    'test-table-1'
+    'test-table-1.jpg'
 ]
 expected_dirs = [
     '33',