diff --git a/advanced_model_merging.py b/advanced_model_merging.py
index 2a69acf..f1d0686 100644
--- a/advanced_model_merging.py
+++ b/advanced_model_merging.py
@@ -12,13 +12,13 @@ class ModelMergeBlockNumber(comfy_extras.nodes_model_merging.ModelMergeBlocks):
         arg_dict["label_emb."] = argument
 
         for i in range(12):
-            arg_dict["input_blocks.{}".format(i)] = argument
+            arg_dict["input_blocks.{}.".format(i)] = argument
 
         for i in range(3):
-            arg_dict["middle_block.{}".format(i)] = argument
+            arg_dict["middle_block.{}.".format(i)] = argument
 
         for i in range(12):
-            arg_dict["output_blocks.{}".format(i)] = argument
+            arg_dict["output_blocks.{}.".format(i)] = argument
 
         arg_dict["out."] = argument