diff --git a/.github/workflows/e2e-test-tune-api.yaml b/.github/workflows/e2e-test-tune-api.yaml
index 6ac0c6e0dfc..2c45d75f99b 100644
--- a/.github/workflows/e2e-test-tune-api.yaml
+++ b/.github/workflows/e2e-test-tune-api.yaml
@@ -24,7 +24,7 @@ jobs:
       
       - name: Install Training Operator SDK
         shell: bash
-        run: pip install kubeflow-training[huggingface]
+        run: pip install -U kubeflow-training[huggingface]
       
       - name: Run e2e test with tune API
         uses: ./.github/workflows/template-e2e-test