diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py
index fc20ead..381cfff 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -92,7 +92,7 @@ def create_trainer() -> Trainer:
 		callbacks =
 		[
 			ModelCheckpoint(
-				monitor = 'l_G',
+				monitor = 'loss_generator',
 				dirpath = output_directory_path,
 				filename = output_file_pattern,
 				every_n_train_steps = 1000,