diff --git a/Assets/Editor/InjectToonFireToShaderGraph.cs b/Assets/Editor/InjectToonFireToShaderGraph.cs
index 3a29caa..a9f5131 100644
--- a/Assets/Editor/InjectToonFireToShaderGraph.cs
+++ b/Assets/Editor/InjectToonFireToShaderGraph.cs
@@ -62,7 +62,11 @@ public static class InjectToonFireToShaderGraph
         foreach (var edge in edges)
         {
             string inputNodeId = (string)edge["m_InputSlot"]["m_Node"]["m_Id"];
-            if (inputNodeId == "5795a95f005b46bbbbf4c4e22a0ae9c3" || inputNodeId == "6c0a4ae63e3c48ad9e08ffbc4cf86bba")
+            string outputNodeId = (string)edge["m_OutputSlot"]["m_Node"]["m_Id"];
+            if (inputNodeId == "5795a95f005b46bbbbf4c4e22a0ae9c3" || 
+                inputNodeId == "6c0a4ae63e3c48ad9e08ffbc4cf86bba" ||
+                inputNodeId == "node_cartoon_fire_custom" ||
+                outputNodeId == "node_cartoon_fire_custom")
             {
                 edgesToRemove.Add(edge);
             }
diff --git a/Assets/Shaders/Fire_Shader.shadergraph b/Assets/Shaders/Fire_Shader.shadergraph
index 57b31f6..48bfba5 100644
--- a/Assets/Shaders/Fire_Shader.shadergraph
+++ b/Assets/Shaders/Fire_Shader.shadergraph
@@ -540,118 +540,6 @@
         "m_SlotId": 1
       }
     },
-    {
-      "m_OutputSlot": {
-        "m_Node": {
-          "m_Id": "0d74e8c23ef1428eaf8c38a4c033c563"
-        },
-        "m_SlotId": 2
-      },
-      "m_InputSlot": {
-        "m_Node": {
-          "m_Id": "node_cartoon_fire_custom"
-        },
-        "m_SlotId": 0
-      }
-    },
-    {
-      "m_OutputSlot": {
-        "m_Node": {
-          "m_Id": "e48170ea86fb4291b3e4198160139b3e"
-        },
-        "m_SlotId": 2
-      },
-      "m_InputSlot": {
-        "m_Node": {
-          "m_Id": "node_cartoon_fire_custom"
-        },
-        "m_SlotId": 1
-      }
-    },
-    {
-      "m_OutputSlot": {
-        "m_Node": {
-          "m_Id": "3f4b2124cb794f8e970aacc0488fc224"
-        },
-        "m_SlotId": 4
-      },
-      "m_InputSlot": {
-        "m_Node": {
-          "m_Id": "node_cartoon_fire_custom"
-        },
-        "m_SlotId": 4
-      }
-    },
-    {
-      "m_OutputSlot": {
-        "m_Node": {
-          "m_Id": "0d74e8c23ef1428eaf8c38a4c033c563"
-        },
-        "m_SlotId": 2
-      },
-      "m_InputSlot": {
-        "m_Node": {
-          "m_Id": "node_cartoon_fire_custom"
-        },
-        "m_SlotId": 0
-      }
-    },
-    {
-      "m_OutputSlot": {
-        "m_Node": {
-          "m_Id": "e48170ea86fb4291b3e4198160139b3e"
-        },
-        "m_SlotId": 2
-      },
-      "m_InputSlot": {
-        "m_Node": {
-          "m_Id": "node_cartoon_fire_custom"
-        },
-        "m_SlotId": 1
-      }
-    },
-    {
-      "m_OutputSlot": {
-        "m_Node": {
-          "m_Id": "3f4b2124cb794f8e970aacc0488fc224"
-        },
-        "m_SlotId": 4
-      },
-      "m_InputSlot": {
-        "m_Node": {
-          "m_Id": "node_cartoon_fire_custom"
-        },
-        "m_SlotId": 4
-      }
-    },
-    {
-      "m_OutputSlot": {
-        "m_Node": {
-          "m_Id": "0d74e8c23ef1428eaf8c38a4c033c563"
-        },
-        "m_SlotId": 2
-      },
-      "m_InputSlot": {
-        "m_Node": {
-          "m_Id": "node_cartoon_fire_custom"
-        },
-        "m_SlotId": 0
-      }
-    },
-    {
-      "m_OutputSlot": {
-        "m_Node": {
-          "m_Id": "e48170ea86fb4291b3e4198160139b3e"
-        },
-        "m_SlotId": 2
-      },
-      "m_InputSlot": {
-        "m_Node": {
-          "m_Id": "node_cartoon_fire_custom"
-        },
-        "m_SlotId": 1
-      }
-    },
     {
       "m_OutputSlot": {
         "m_Node": {