diff --git a/stream_diffusion_nodes.py b/stream_diffusion_nodes.py
index 9d766fc..03db7dd 100644
--- a/stream_diffusion_nodes.py
+++ b/stream_diffusion_nodes.py
@@ -74,12 +74,19 @@ class StreamDiffusionPipeline(ControlNodeBase):
             "sigmas": None,
             "stock_noise": None,  # For RCFG
             "timesteps": None,
+            "current_buffer_size": None,  # Track buffer size
         })
         
         # Get latent tensor from dict
         samples = latents["samples"]  # Shape: [1, 4, H, W]
         device = samples.device
         
+        # Check if buffer size changed
+        if state["current_buffer_size"] != frame_buffer_size:
+            state["latent_buffer"] = None  # Force reinitialization
+            state["current_buffer_size"] = frame_buffer_size
+            state["current_idx"] = 0
+        
         # Initialize scheduler and timesteps if needed
         if state["sigmas"] is None:
             # Calculate sigmas using ComfyUI's scheduler system