diff --git a/src/more_math/NoiseMathNode.py b/src/more_math/NoiseMathNode.py
index b1282b1..27f6722 100644
--- a/src/more_math/NoiseMathNode.py
+++ b/src/more_math/NoiseMathNode.py
@@ -141,7 +141,7 @@ class NoiseMathNode:
         H = getIndexTensorAlongDim(input_latent["samples"], 3)
         C = getIndexTensorAlongDim(input_latent["samples"], 1)
 
-        variables = {'a': self.va.generate_noise(input_latent), 'b': self.vb, 'c': self.vc, 'd': self.vd, 'w': self.vw, 'x': self.vx, 'y': self.vy, 'z': self.vz,'B':B,'X':W,'Y':H,'C':C,'W':input_latent["samples"].shape[2],'H':input_latent["samples"].shape[3]}
+        variables = {'a': self.va.generate_noise(input_latent), 'b': self.vb, 'c': self.vc, 'd': self.vd, 'w': self.vw, 'x': self.vx, 'y': self.vy, 'z': self.vz,'B':B,'X':W,'Y':H,'C':C,'W':input_latent["samples"].shape[2],'H':input_latent["samples"].shape[3],'I':input_latent["samples"]}
         input_stream = InputStream(self.expr)
         lexer = MathExprLexer(input_stream)
         stream = CommonTokenStream(lexer)