diff --git a/detector/modeler.py b/detector/modeler.py
index 49fd8db..8cb121f 100644
--- a/detector/modeler.py
+++ b/detector/modeler.py
@@ -187,7 +187,7 @@ def train(
                 # logits = logits.detach().cpu().numpy()
                 # label_ids = b_labels.detach().cpu().numpy()
 
-                total_eval_accuracy += self.accuracy(logits, label_ids)
+                total_eval_accuracy += self.accuracy(logits, b_labels)
 
             avg_val_accuracy = total_eval_accuracy / len(validation_dataloader)
             print(f'Validation Accuracy: {avg_val_accuracy:.4f}')
@@ -475,7 +475,7 @@ def train(
                 # logits = logits.detach().cpu().numpy()
                 # label_ids = b_labels.detach().cpu().numpy()
 
-                total_eval_accuracy += self.accuracy(logits, label_ids)
+                total_eval_accuracy += self.accuracy(logits, b_labels)
 
             avg_val_accuracy = total_eval_accuracy / len(validation_dataloader)
             print(f'Validation Accuracy: {avg_val_accuracy:.4f}')