diff --git a/vllm_omni/entrypoints/omni_llm.py b/vllm_omni/entrypoints/omni_llm.py
index 1312af335..cdfe5d53c 100644
--- a/vllm_omni/entrypoints/omni_llm.py
+++ b/vllm_omni/entrypoints/omni_llm.py
@@ -15,7 +15,7 @@ from vllm.logger import init_logger
 import vllm.envs as envs
 
 from vllm_omni.entrypoints.utils import load_stage_configs_from_model
-from vllm_omni.entrypoints.stage import OmniStage
+from vllm_omni.entrypoints.omni_stage import OmniStage
 from vllm_omni.engine.arg_utils import OmniEngineArgs
 from vllm_omni.engine.output_processor import MultimodalOutputProcessor
 from vllm_omni.engine.processor import OmniProcessor
diff --git a/vllm_omni/entrypoints/stage.py b/vllm_omni/entrypoints/omni_stage.py
similarity index 100%
rename from vllm_omni/entrypoints/stage.py
rename to vllm_omni/entrypoints/omni_stage.py