diff --git a/image_ref/main.py b/image_ref/main.py
index 80ee5e1a1d1071e8debd2fd58580d72926ba870d..b9b861112b4e15bc4102cc3cf8e5dfdcc439eaf5 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -164,10 +164,13 @@ def make_prediction_duo(model, data, f_name, f_name2):
         confidence_pred_list[specie].append(output[:,0].data.cpu().numpy())
         #Mono class output (only most postive paire)
         output = torch.argmax(output[:,0])
+        label = torch.argmin(label)
 
+        print(output, label)
         y_pred.extend(output)
 
         label = torch.argmin(label)
+        print(output, label)
         y_true.extend(label)  # Save Truth
     # constant for classes