diff --git a/infod_sample.py b/infod_sample.py
index b8844b2..6b257d5 100644
--- a/infod_sample.py
+++ b/infod_sample.py
@@ -15,6 +15,8 @@
 from utils import *
 from compression import *
 from decompression import *
+from PIL import ImageFile
+ImageFile.LOAD_TRUNCATED_IMAGES = True
 
 
 class InfoDrop(Attack):