diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py
index 8e8b086..9e02685 100644
--- a/embedding_converter/src/training.py
+++ b/embedding_converter/src/training.py
@@ -94,7 +94,7 @@ def create_trainer() -> Trainer:
 		callbacks =
 		[
 			ModelCheckpoint(
-				monitor = 'train_loss',
+				monitor = 'loss_training',
 				dirpath = output_directory_path,
 				filename = output_file_pattern,
 				every_n_epochs = 10,