diff --git a/reIDfolder.py b/reIDfolder.py
index b48c0f5..39ba5b9 100755
--- a/reIDfolder.py
+++ b/reIDfolder.py
@@ -37,7 +37,7 @@ def _get_neg_sample(self, target):
         neg_index = np.argwhere(self.targets != target)
         neg_index = neg_index.flatten()
         rand = random.randint(0,len(neg_index)-1)
-        return self.samples[neg_index[rand]]
+        return self.samples[neg_index[rand]][0]
 
     def __getitem__(self, index):
         path, target = self.samples[index]