diff --git a/modules/ui.py b/modules/ui.py
index 79ccd240c62..08a5531d8f8 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1980,6 +1980,7 @@ def sagemaker_train_embedding(
                     
                 def sagemaker_train_hypernetwork(
                         username,
+                        sd_model_checkpoint,
                         new_hypernetwork_name,
                         new_hypernetwork_sizes,
                         new_hypernetwork_layer_structure,