diff --git a/helpers/legacy/validation.py b/helpers/legacy/validation.py
index 4449c1ef..0336236a 100644
--- a/helpers/legacy/validation.py
+++ b/helpers/legacy/validation.py
@@ -86,6 +86,7 @@ def log_validations(
     unet,
     args,
     validation_prompts,
+    validation_shortnames,
     global_step,
     resume_global_step,
     step,