diff --git a/tensorrt_llm/llmapi/disagg_utils.py b/tensorrt_llm/llmapi/disagg_utils.py
index d1851364d7c5..94510bd8f62f 100644
--- a/tensorrt_llm/llmapi/disagg_utils.py
+++ b/tensorrt_llm/llmapi/disagg_utils.py
@@ -15,7 +15,7 @@
 __all__ = [
     'ServerConfig',
     'parse_disagg_config_file',
-    'extract_server_configs',
+    'extract_disagg_cfg',
     'split_world_comm',
     'get_usage_tokens_from_ctx',
     'rewrite_usage_info_from_ctx',