diff --git a/adv_control/control_sparsectrl.py b/adv_control/control_sparsectrl.py
index e10e08c..26c3956 100644
--- a/adv_control/control_sparsectrl.py
+++ b/adv_control/control_sparsectrl.py
@@ -146,7 +146,8 @@ class SparseModelPatcher(ModelPatcher):
 
         n.object_patches = self.object_patches.copy()
         n.model_options = copy.deepcopy(self.model_options)
-        n.model_keys = self.model_keys
+        if hasattr(n, "model_keys"):
+            n.model_keys = self.model_keys
         if hasattr(n, "backup"):
             self.backup = n.backup
         if hasattr(n, "object_patches_backup"):
diff --git a/pyproject.toml b/pyproject.toml
index d3d8082..6b0be65 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-advanced-controlnet"
 description = "Nodes for scheduling ControlNet strength across timesteps and batched latents, as well as applying custom weights and attention masks."
-version = "1.0.3"
+version = "1.0.4"
 license = "LICENSE"
 dependencies = []