diff --git a/image_ref/main.py b/image_ref/main.py
index b4b6e22d2e747f0be133a46a27e6d4247086964e..12ebb403bd5520d65795295c4f856dda55a90d50 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -46,10 +46,10 @@ def test_duo(model, data_test, loss_function, epoch):
         param.requires_grad = False
 
     for imaer,imana, img_ref, label in data_test:
-        imaer.permute(0,1)
-        imana.permute(0,1)
-        img_ref.permute(0,1)
-        label.permute(0,1)
+        imaer.transpose(0,1)
+        imana.transpose(0,1)
+        img_ref.transpose(0,1)
+        label.transpose(0,1)
         label = label.long()
         if torch.cuda.is_available():
             imaer = imaer.cuda()