diff --git a/nemo/export/trt_llm/nemo_utils.py b/nemo/export/trt_llm/nemo_utils.py
index da53890d844e..e3c226ef639e 100644
--- a/nemo/export/trt_llm/nemo_utils.py
+++ b/nemo/export/trt_llm/nemo_utils.py
@@ -197,9 +197,7 @@ def nemo_llm_to_model_config(
 
 
 def to_word_list_format(
-    word_dict: List[List[str]], 
-    tokenizer=None, 
-    ref_str="Green tea icecream",
+    word_dict: List[List[str]], tokenizer=None, ref_str="Green tea icecream",
 ):
     '''
     format of word_dict