diff --git a/nodes.py b/nodes.py
index 0b174fe..df190b3 100644
--- a/nodes.py
+++ b/nodes.py
@@ -118,7 +118,8 @@ def teacache_flux_forward(
                                                             "vec": vec,
                                                             "pe": pe,
                                                             "attn_mask": attn_mask},
-                                                            {"original_block": block_wrap})
+                                                            {"original_block": block_wrap,
+                                                             "transformer_options": transformer_options})
                     txt = out["txt"]
                     img = out["img"]
                 else:
@@ -161,7 +162,8 @@ def teacache_flux_forward(
                                                             "vec": vec,
                                                             "pe": pe,
                                                             "attn_mask": attn_mask}, 
-                                                            {"original_block": block_wrap})
+                                                            {"original_block": block_wrap,
+                                                             "transformer_options": transformer_options})
                     img = out["img"]
                 else:
                     img = block(img, vec=vec, pe=pe, attn_mask=attn_mask)
diff --git a/pyproject.toml b/pyproject.toml
index 8c20eae..532215c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "teacache"
 description = "Unofficial implementation of [ali-vilab/TeaCache](https://github.com/ali-vilab/TeaCache) for ComfyUI"
-version = "1.4.3"
+version = "1.4.4"
 license = {file = "LICENSE"}
 
 [project.urls]