diff --git a/utils/datasets.py b/utils/datasets.py
index 169a518..7ed8f5a 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -457,7 +457,7 @@ class LoadImagesAndLabels(Dataset):  # for training/testing
                     l = np.zeros((0, 5), dtype=np.float32)
                 x[img] = [l, shape]
             except Exception as e:
-                x[img] = None
+                x[img] = [None, None]
                 print('WARNING: %s: %s' % (img, e))
 
         x['hash'] = get_hash(self.label_files + self.img_files)