diff --git a/tests/utils.py b/tests/utils.py
index b7d177b6..cdd32293 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -157,6 +157,8 @@ def run_prediction(dataloader, trained_model, dp=False, min_acc=0.50):
         acc = torch.mean(acc).item()
 
     else:
+        import pdb
+        pdb.set_trace()
         y_hat = trained_model(x)
 
         # acc