diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index 163b45eb..ad89b052 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -36,5 +36,5 @@ MAX_TOKENS = {
     'bedrock/anthropic.claude-3-haiku-20240307-v1:0': 100000,
     'groq/llama3-8b-8192': 8192,
     'groq/llama3-70b-8192': 8192,
-    'llama/llama3': 4096,
+    'ollama/llama3': 4096,
 }