diff --git a/nodes.py b/nodes.py
index 51ccceb..06a2ed7 100644
--- a/nodes.py
+++ b/nodes.py
@@ -2394,8 +2394,10 @@ class WanVideoSampler:
             sample_scheduler = FlowMatchLCMScheduler(shift=shift, use_beta_sigmas=(scheduler == 'lcm/beta'))
             sample_scheduler.set_timesteps(steps, device=device, sigmas=sigmas.tolist() if sigmas is not None else None)
         elif 'flowmatch_causvid' in scheduler:
+            denoising_list = [999, 934, 862, 756, 603, 410, 250, 140, 74]
             sample_scheduler = FlowMatchScheduler(num_inference_steps=steps, shift=shift, sigma_min=0, extra_one_step=True)
-            sample_scheduler.timesteps = torch.tensor([999, 934, 862, 756, 603, 410, 250, 140, 74])[:steps].to(device)
+            sample_scheduler.timesteps = torch.tensor(denoising_list)[:steps].to(device)
+            sample_scheduler.sigmas = torch.cat([sample_scheduler.timesteps / 1000, torch.tensor([0.0], device=device)])
         
         if timesteps is None:
             timesteps = sample_scheduler.timesteps