diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index da54221..e57f46a 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_cuda
         with:
           image: ghcr.io/tue-robotics/tue-env-ros-noetic-cu12.1.1-cudnn8-devel
           package: ${{ github.event.repository.name }}