diff --git a/crates/llm-ls/src/backend.rs b/crates/llm-ls/src/backend.rs
index 478a94c..93c58e5 100644
--- a/crates/llm-ls/src/backend.rs
+++ b/crates/llm-ls/src/backend.rs
@@ -125,8 +125,8 @@ enum OllamaAPIResponse {
     Error(APIError),
 }
 
-fn build_ollama_headers() -> HeaderMap {
-    HeaderMap::new()
+fn build_ollama_headers(api_token: Option<&String>, ide: Ide) -> Result<HeaderMap> {
+    build_api_headers(api_token, ide)
 }
 
 fn parse_ollama_text(text: &str) -> Result<Vec<Generation>> {
@@ -247,7 +247,7 @@ pub(crate) fn build_headers(
     match backend {
         Backend::HuggingFace { .. } => build_api_headers(api_token, ide),
         Backend::LlamaCpp { .. } => build_llamacpp_headers(api_token, ide),
-        Backend::Ollama { .. } => Ok(build_ollama_headers()),
+        Backend::Ollama { .. } => build_ollama_headers(api_token, ide),
         Backend::OpenAi { .. } => build_openai_headers(api_token, ide),
         Backend::Tgi { .. } => build_tgi_headers(api_token, ide),
     }