diff --git a/wbia_miew_id/train.py b/wbia_miew_id/train.py
index bfd6cbc..a2bd98f 100644
--- a/wbia_miew_id/train.py
+++ b/wbia_miew_id/train.py
@@ -46,7 +46,7 @@ def run_fn(self, model, train_loader, valid_loader, criterion, optimizer, schedu
                 print('Group average score: ', valid_score)
             else:
                 print('Evaluating on full test set')
-                valid_score = eval_fn(valid_loader, model, device, use_wandb=use_wandb)
+                valid_score, valid_cmc = eval_fn(valid_loader, model, device, use_wandb=use_wandb)
                 print('Valid score: ', valid_score)
 
             if valid_score > best_score: