diff --git a/src/backend/langflow/components/models/CohereModel.py b/src/backend/langflow/components/models/CohereModel.py
index af76e1adc..3a25a3593 100644
--- a/src/backend/langflow/components/models/CohereModel.py
+++ b/src/backend/langflow/components/models/CohereModel.py
@@ -1,3 +1,6 @@
+from typing import Optional
+
+
 from langchain_community.chat_models.cohere import ChatCohere
 from pydantic.v1 import SecretStr