diff --git a/extensions-builtin/sd_forge_mod_guidance/lib_modulation/anima_patch.py b/extensions-builtin/sd_forge_mod_guidance/lib_modulation/anima_patch.py
index d4493343..cbf85227 100644
--- a/extensions-builtin/sd_forge_mod_guidance/lib_modulation/anima_patch.py
+++ b/extensions-builtin/sd_forge_mod_guidance/lib_modulation/anima_patch.py
@@ -75,7 +75,7 @@ def register_modulation_wrapper(
         "end_layer": int(end_layer),
     }
 
-    assert isinstance(model_patcher.model.diffusion_model, Anima)
+    assert model_patcher.model.config.huggingface_repo.endswith("Anima")
 
     global ORIG_FORWARD
     ORIG_FORWARD = Anima.forward