diff --git a/llama.h b/llama.h
index 6b348ac38..cfc2ee873 100644
--- a/llama.h
+++ b/llama.h
@@ -390,9 +390,10 @@ extern "C" {
     LLAMA_API uint32_t llama_n_ubatch   (const struct llama_context * ctx);
     LLAMA_API uint32_t llama_n_seq_max  (const struct llama_context * ctx);
 
+    LLAMA_API enum llama_pooling_type llama_pooling_type(const struct llama_context * ctx);
+
     LLAMA_API enum llama_vocab_type   llama_vocab_type  (const struct llama_model   * model);
     LLAMA_API enum llama_rope_type    llama_rope_type   (const struct llama_model   * model);
-    LLAMA_API enum llama_pooling_type llama_pooling_type(const struct llama_context * model);
 
     LLAMA_API int32_t llama_n_vocab    (const struct llama_model * model);
     LLAMA_API int32_t llama_n_ctx_train(const struct llama_model * model);