diff --git a/src/xe_forge/core/executor.py b/src/xe_forge/core/executor.py
index 8debed4..864e93e 100644
--- a/src/xe_forge/core/executor.py
+++ b/src/xe_forge/core/executor.py
@@ -269,6 +269,8 @@ def _check_correctness(
                 return False
 
             if original_model is not None:
+                # Run in eval mode instead of training mode.
+                original_model.eval()
                 original_model = original_model.to(device)
                 if dtype is not None:
                     original_model = original_model.to(dtype)
@@ -289,6 +291,8 @@ def _check_correctness(
                 return False
 
             if optimized_model is not None:
+                # Run in eval mode instead of training mode.
+                optimized_model.eval()
                 optimized_model = optimized_model.to(device)
                 if dtype is not None:
                     optimized_model = optimized_model.to(dtype)