diff --git a/nodes_sampler.py b/nodes_sampler.py
index a44073d..fab7dfd 100644
--- a/nodes_sampler.py
+++ b/nodes_sampler.py
@@ -5,10 +5,10 @@ from tqdm import tqdm
 import inspect
 from PIL import Image
 from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
-
+from .wanvideo.schedulers.fm_solvers import get_sampling_sigmas, retrieve_timesteps
 from .wanvideo.modules.model import rope_params
 from .custom_linear import remove_lora_from_module, set_lora_params, _replace_linear
-from .wanvideo.schedulers import get_scheduler, get_sampling_sigmas, retrieve_timesteps, scheduler_list
+from .wanvideo.schedulers import get_scheduler, scheduler_list
 from .gguf.gguf import set_lora_params_gguf
 from .multitalk.multitalk import timestep_transform, add_noise
 from .utils import(log, print_memory, apply_lora, fourier_filter, optimized_scale, setup_radial_attention,