diff --git a/tests/utils.py b/tests/utils.py
index cdd32293..16a6a19f 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -144,6 +144,8 @@ def load_model(exp, root_weights_dir, module_class=LightningTemplateModel):
 
 
 def run_prediction(dataloader, trained_model, dp=False, min_acc=0.50):
+    trained_model.eval()
+
     # run prediction on 1 batch
     for batch in dataloader:
         break