diff --git a/action.yml b/action.yml
index 7872904..41385f6 100644
--- a/action.yml
+++ b/action.yml
@@ -24,7 +24,7 @@ runs:
   using: composite
   steps:
     - name: Setup ollama
-      uses: ai-action/setup-ollama@v1
+      uses: ai-action/setup-ollama@v2
 
     - name: Checkout repository
       uses: actions/checkout@v6