diff --git a/private_gpt/ui/ui.py b/private_gpt/ui/ui.py
index 2d942a42..00cdcd56 100644
--- a/private_gpt/ui/ui.py
+++ b/private_gpt/ui/ui.py
@@ -418,7 +418,7 @@ class PrivateGptUi:
                         inputs=system_prompt_input,
                     )
 
-                    def get_model_label() -> str:
+                    def get_model_label() -> str | None:
                         """Get model label from llm mode setting YAML.
 
                         Raises: