diff --git a/learning_loop_node/tests/test_downloader.py b/learning_loop_node/tests/test_downloader.py
index 284e2898..bf2e10e8 100644
--- a/learning_loop_node/tests/test_downloader.py
+++ b/learning_loop_node/tests/test_downloader.py
@@ -2,7 +2,7 @@
 import shutil
 
 from learning_loop_node.data_classes import Context
-from learning_loop_node.data_exchanger import DataExchanger
+from learning_loop_node.data_exchanger import DataExchanger, check_jpeg
 from learning_loop_node.globals import GLOBALS
 
 from . import test_helper
@@ -67,5 +67,5 @@ async def test_removal_of_corrupted_images(data_exchanger: DataExchanger):
 
     await data_exchanger.delete_corrupt_images('/tmp/img_folder')
 
-    assert len(os.listdir('/tmp/img_folder')) == num_images
+    assert len(os.listdir('/tmp/img_folder')) == num_images if check_jpeg else num_images - 1
     shutil.rmtree('/tmp/img_folder', ignore_errors=True)