diff --git a/image_ref/dataset_ref.py b/image_ref/dataset_ref.py
index 738820aa194f56a2f5e29188f7e2fc8c4fb26831..ae54025638fe176a54d74cde0b6610666bdf63c9 100644
--- a/image_ref/dataset_ref.py
+++ b/image_ref/dataset_ref.py
@@ -225,8 +225,8 @@ class ImageFolderDuo_Batched(data.Dataset):
 
         batched_im_ref = torch.concat(img_refs,dim=0)
         batched_label = torch.tensor(label_refs)
-        batched_imgAER = imgAER.Tensor.repeat(len(self.classes))
-        batched_imgANA = imgANA.Tensor.repeat(len(self.classes))
+        batched_imgAER = imgAER.repeat(len(self.classes))
+        batched_imgANA = imgANA.repeat(len(self.classes))
 
         return batched_imgAER, batched_imgANA, batched_im_ref, batched_label