diff --git a/tests/unittest/dynamo/test_imports.py b/tests/unittest/dynamo/test_imports.py
index 981bf90f9df5..8e29ff34127a 100644
--- a/tests/unittest/dynamo/test_imports.py
+++ b/tests/unittest/dynamo/test_imports.py
@@ -51,6 +51,7 @@
     ("tensorrt_llm.tokenizer", "TOKENIZER_ALIASES"),
     ("tensorrt_llm.llmapi.llm_args", "KvCacheConnectorConfig"),
     ("tensorrt_llm.llmapi.llm_args", "TorchLlmArgs"),
+    ("tensorrt_llm.llmapi.llm_args", "BaseLlmArgs"),
     ("tensorrt_llm.llmapi.llm_utils", "update_llm_args_with_extra_options"),
     ("tensorrt_llm.llmapi.tokenizer", "tokenizer_factory"),
     ("tensorrt_llm.llmapi.disagg_utils", "get_global_disagg_request_id"),