diff --git a/comfy_extras/nodes_custom_sampler.py b/comfy_extras/nodes_custom_sampler.py
index a5d791d9..56cb505e 100644
--- a/comfy_extras/nodes_custom_sampler.py
+++ b/comfy_extras/nodes_custom_sampler.py
@@ -320,7 +320,7 @@ class Noise_EmptyNoise:
 
     def generate_noise(self, input_latent):
         latent_image = input_latent["samples"]
-        return torch.zeros(shape, dtype=latent_image.dtype, layout=latent_image.layout, device="cpu")
+        return torch.zeros(latent_image.shape, dtype=latent_image.dtype, layout=latent_image.layout, device="cpu")
 
 
 class Noise_RandomNoise:
@@ -477,7 +477,7 @@ class DisableNoise:
     FUNCTION = "get_noise"
     CATEGORY = "sampling/custom_sampling/noise"
 
-    def get_noise(self, noise_seed):
+    def get_noise(self):
         return (Noise_EmptyNoise(),)