diff --git a/vllm_omni/entrypoints/async_omni_llm.py b/vllm_omni/entrypoints/async_omni_llm.py
index 7cc138efb..730001173 100644
--- a/vllm_omni/entrypoints/async_omni_llm.py
+++ b/vllm_omni/entrypoints/async_omni_llm.py
@@ -110,7 +110,7 @@ class AsyncOmniLLM(EngineClient):
         # Optional file handler for orchestrator
         self._log_file = log_file
         if self._log_file:
-            remove_old_logs(self._log_file, len(self.stage_list))
+            remove_old_logs(self._log_file, len(self.stage_configs))
             configure_orchestrator_logger(logger, self._log_file)
 
         self._stats_file, self._overall_stats_file = init_stats_paths(self._enable_stats, self._log_file)
diff --git a/vllm_omni/entrypoints/omni_llm.py b/vllm_omni/entrypoints/omni_llm.py
index c03d3ce2f..dccc380ae 100644
--- a/vllm_omni/entrypoints/omni_llm.py
+++ b/vllm_omni/entrypoints/omni_llm.py
@@ -97,7 +97,7 @@ class OmniLLM:
         # Optional file handler for orchestrator
         self._log_file = log_file
         if self._log_file:
-            remove_old_logs(self._log_file, len(self.stage_list))
+            remove_old_logs(self._log_file, len(self.stage_configs))
             configure_orchestrator_logger(logger, self._log_file)
 
         self._stats_file, self._overall_stats_file = init_stats_paths(self._enable_stats, self._log_file)