diff --git a/TrainerClass.py b/TrainerClass.py
index 9012a38..1928ba4 100644
--- a/TrainerClass.py
+++ b/TrainerClass.py
@@ -578,7 +578,7 @@ class eNoseTrainer:
                 df.to_excel(f"{model_file}.search.xlsx", index=False)
 
                 self.logger.info(f"Training Model {model_id} with {study.best_params}")
-                early_stopping = keras.callbacks.EarlyStopping(monitor='val_loss', patience=5, restore_best_weights=True)
+                early_stopping = keras.callbacks.EarlyStopping(monitor='loss', patience=5, restore_best_weights=True)
                 best_model.fit(X_train, Y_train, epochs=epochs, batch_size=best_batch_size, verbose=1, callbacks=[early_stopping])
 
                 best_model.save(f"{model_file}.keras")