diff --git a/compare_images.py b/compare_images.py
index c0b7e42..0017589 100644
--- a/compare_images.py
+++ b/compare_images.py
@@ -55,7 +55,7 @@ def choose_images():
 
     # REPLACE "USER" WITH YOUR USERNAME
     # Check if there are at least two images
-    if len(os.listdir('/home/USER/klipper-camera-watchdog/Image-files/*.jpg')) > 1:
+    if len(os.listdir('/home/USER/klipper-camera-watchdog/Image-files/')) > 1:
         # Sort image paths by creation time (newest first)
         image_paths = sorted(glob.glob('/home/USER/klipper-camera-watchdog/Image-files/*.jpg'), key=os.path.getctime, reverse=True)