diff --git a/fbcache_nodes.py b/fbcache_nodes.py
index 4aaeab8..3b6da43 100644
--- a/fbcache_nodes.py
+++ b/fbcache_nodes.py
@@ -121,11 +121,11 @@ def validate_use_cache(use_cached):
         model = model.clone()
         diffusion_model = model.get_model_object(object_to_patch)
 
-        if diffusion_model.__class__.__name__ in ("UNetModel", "FLUX"):
+        if diffusion_model.__class__.__name__ in ("UNetModel", "Flux"):
 
             if diffusion_model.__class__.__name__ == "UNetModel":
                 create_patch_function = first_block_cache.create_patch_unet_model__forward
-            elif diffusion_model.__class__.__name__ == "FLUX":
+            elif diffusion_model.__class__.__name__ == "Flux":
                 create_patch_function = first_block_cache.create_patch_flux_forward_orig
             else:
                 raise ValueError(
diff --git a/first_block_cache.py b/first_block_cache.py
index 783f060..0d75f57 100644
--- a/first_block_cache.py
+++ b/first_block_cache.py
@@ -561,7 +561,7 @@ def call_remaining_blocks(self, blocks_replace, control, img, txt, vec,
                               pe, attn_mask):
         original_hidden_states = img
 
-        for i, block in enumerate(self.double_block):
+        for i, block in enumerate(self.double_blocks):
             if i < 1:
                 continue
             if ("double_block", i) in blocks_replace:
diff --git a/pyproject.toml b/pyproject.toml
index c7a0d02..c1a1114 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "wavespeed"
 description = "The all in one inference optimization solution for ComfyUI, universal, flexible, and fast."
-version = "1.1.2"
+version = "1.1.3"
 license = {file = "LICENSE"}
 
 [project.urls]