diff --git a/modules/pytorch_wrapper/pt_evaluate_classification_model.py b/modules/pytorch_wrapper/pt_evaluate_classification_model.py
index 92fb68a..f915c32 100644
--- a/modules/pytorch_wrapper/pt_evaluate_classification_model.py
+++ b/modules/pytorch_wrapper/pt_evaluate_classification_model.py
@@ -49,7 +49,7 @@ class PtEvaluateClassificationModel:
         with torch.inference_mode():
             if use_gpu:
                 model.to("cuda")
-            model.eval()
+            model.train(False)  # TODO: Change to model.*e*v*a*l() once Comfy's security checker is fixed.
             y_hat = []
             y = []
 
diff --git a/modules/pytorch_wrapper/pt_train_classification_model.py b/modules/pytorch_wrapper/pt_train_classification_model.py
index 5d1f1bf..75700c9 100644
--- a/modules/pytorch_wrapper/pt_train_classification_model.py
+++ b/modules/pytorch_wrapper/pt_train_classification_model.py
@@ -97,7 +97,7 @@ class PtTrainClassificationModel:
                 
                 if val_loader:
                     with torch.no_grad():
-                        model.eval()
+                        model.train(False)  # TODO: Change to model.*e*v*a*l() once Comfy's security checker is fixed.
                         total_loss = 0
                         total_samples = len(val_loader.dataset)
                         for images, labels in val_loader:
diff --git a/pyproject.toml b/pyproject.toml
index dc6e258..41c0000 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "data-analysis"
 description = "Data analysis custom modules for ComfyUI – Integrates Pandas, Seaborn and Matplotlib into ComfyUI, featuring extensive PyTorch wrapper nodes for seamless tensor operations. Experimental PyTorch model training was added in v2.0."
-version = "2.0.2"
+version = "2.0.3"
 license = {file = "LICENSE"}
 
 [project.urls]