diff --git a/train_sd21.py b/train_sd21.py
index ef4908cd..875cf210 100644
--- a/train_sd21.py
+++ b/train_sd21.py
@@ -998,6 +998,7 @@ def collate_fn(batch):
                 unet,
                 args,
                 validation_prompts,
+                validation_shortnames,
                 global_step,
                 resume_global_step,
                 step,
diff --git a/train_sdxl.py b/train_sdxl.py
index 8e469045..b1423155 100644
--- a/train_sdxl.py
+++ b/train_sdxl.py
@@ -1177,6 +1177,7 @@ def collate_fn(batch):
                 unet,
                 args,
                 validation_prompts,
+                validation_shortnames,
                 global_step,
                 resume_global_step,
                 step,