diff --git a/vllm_omni/entrypoints/omni_llm.py b/vllm_omni/entrypoints/omni_llm.py
index 51fb18ebb..dd9791c5a 100644
--- a/vllm_omni/entrypoints/omni_llm.py
+++ b/vllm_omni/entrypoints/omni_llm.py
@@ -1,5 +1,6 @@
 import multiprocessing as mp
 import os
+import sys
 import time
 import uuid
 from collections.abc import Sequence