diff --git a/research/cv/FaceRecognitionForTracking/eval.py b/research/cv/FaceRecognitionForTracking/eval.py
index aba67c8183b1a5ccf50ae1fc959f70e1525a2259..ceee98c3b7aeadd72959ab5b67fcade160f430ba 100644
--- a/research/cv/FaceRecognitionForTracking/eval.py
+++ b/research/cv/FaceRecognitionForTracking/eval.py
@@ -104,7 +104,7 @@ def load_images(paths, batch_size=128):
         im = resize(im)
         img = np.array(im)
         ts = transform(img)
-        ll.append(ts[0])
+        ll.append(ts)
         if len(ll) == batch_size:
             yield np.stack(ll, axis=0)
             ll.clear()
diff --git a/research/cv/FaceRecognitionForTracking/preprocess.py b/research/cv/FaceRecognitionForTracking/preprocess.py
index 1401348119e1e7c2abf30e99a5ae466e045164f0..b4c813207e6bbc47632e63ae732feb8614b0b4bd 100644
--- a/research/cv/FaceRecognitionForTracking/preprocess.py
+++ b/research/cv/FaceRecognitionForTracking/preprocess.py
@@ -34,7 +34,7 @@ def load_images(paths, batch_size=1):
         im = resize(im)
         img = np.array(im)
         ts = transform(img)
-        ll.append(ts[0])
+        ll.append(ts)
         if len(ll) == batch_size:
             yield np.stack(ll, axis=0)
             ll.clear()