diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 0f287553..1727a00f 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -8,7 +8,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: TUe CI
-        uses: tue-robotics/tue-env/ci/main@master
+        uses: tue-robotics/tue-env/ci/main@docker_cuda2
         with:
           package: ${{ github.event.repository.name }}
           ssh-key: ${{ secrets.SSH_DEPLOY_KEY }}