diff --git a/lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/cohere.py b/lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/cohere.py
index 285e77992..d8a0fa700 100644
--- a/lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/cohere.py
+++ b/lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/cohere.py
@@ -4,7 +4,7 @@
 from langchain.prompts.prompt import PromptTemplate
 
 from ..base import ModelAdapter
-from .base import get_guardrails
+from .base import get_guardrails, Bedrock
 from genai_core.registry import registry
 
 
@@ -62,5 +62,5 @@ def get_prompt(self):
 
 # Register the adapter
 registry.register(
-    r"^bedrock\.cohere\.command-(text|light-text|r).*", BedrockCohereCommandAdapter
+    r"^bedrock\.cohere\.command-(text|light-text).*", BedrockCohereCommandAdapter
 )