diff --git a/vllm_omni/entrypoints/omni_llm.py b/vllm_omni/entrypoints/omni_llm.py
index c52e2e969..c1b12a1b2 100644
--- a/vllm_omni/entrypoints/omni_llm.py
+++ b/vllm_omni/entrypoints/omni_llm.py
@@ -34,7 +34,7 @@ from vllm_omni.entrypoints.log_utils import (
     init_stats_paths,
     OrchestratorMetrics,
 )
-from vllm_omni.entrypoints.pipeline_utils import (
+from vllm_omni.entrypoints.stage_utils import (
     maybe_load_from_ipc as _load,
     encode_for_ipc as _encode,
     serialize_obj as _ser,
diff --git a/vllm_omni/entrypoints/omni_stage.py b/vllm_omni/entrypoints/omni_stage.py
index 6ec93878e..8f95cf62e 100644
--- a/vllm_omni/entrypoints/omni_stage.py
+++ b/vllm_omni/entrypoints/omni_stage.py
@@ -11,7 +11,7 @@ import logging
 import multiprocessing as mp
 from typing import Any, Dict, List, Optional, Union
 
-from vllm_omni.entrypoints.pipeline_utils import (
+from vllm_omni.entrypoints.stage_utils import (
     _to_dict,
     set_stage_gpu_devices,
     maybe_load_from_ipc_with_metrics,