diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index 6058078..8a99e68 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -3,7 +3,7 @@ name: OpenSplat (Docker CUDA)
 on:
   push:
     branches:
-      - docker-ubuntu1804
+      - main
   pull_request:
     types: [ assigned, opened, synchronize, reopened ]
   release:
@@ -18,7 +18,7 @@ jobs:
       matrix:
         os: [ubuntu-22.04] # [ubuntu-22.04, ubuntu-20.04, ubuntu-18.04]
         arch: [x64] # [x64, x86]
-        ubuntu-version: [20.04] # [18.04, 20.04, 22.04]
+        ubuntu-version: [22.04, 20.04] # [18.04, 20.04, 22.04]
         torch-version: [2.1.2] # [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.1.1, 2.1.2, 2.2.0, 2.2.1]
         cuda-version: [11.8.0] # [12.3.1, 12.1.1, 11.8.0, 11.7.1, 11.6.2, 11.5.2,11.4.4, 11.3.1, 11.2.2, 11.1.1, 11.0.3, cpu]
         cmake-build-type: [Release] # [Debug, ClangTidy]
diff --git a/.github/workflows/hip.yml b/.github/workflows/hip.yml
index 962631d..6b2de65 100644
--- a/.github/workflows/hip.yml
+++ b/.github/workflows/hip.yml
@@ -93,7 +93,7 @@ jobs:
       - name: Install CUDA
         if: ${{ steps.cuda-cache.outputs.cache-hit != 'true' }}
         run: |
-          bash .github/workflows/cuda/${{ runner.os }}.sh ${{ matrix.cuda-version }}
+          bash .github/workflows/cuda/${{ runner.os }}.sh ${{ matrix.os }} ${{ matrix.cuda-version }}
 
       - name: Restore ROCm Cache
         uses: actions/cache@v4
diff --git a/.github/workflows/ubuntu.yml b/.github/workflows/ubuntu.yml
index fcb91b4..973278f 100644
--- a/.github/workflows/ubuntu.yml
+++ b/.github/workflows/ubuntu.yml
@@ -66,7 +66,7 @@ jobs:
       - name: Install CUDA
         if: ${{ steps.cuda-cache.outputs.cache-hit != 'true' }}
         run: |
-          bash .github/workflows/cuda/${{ runner.os }}.sh ${{ matrix.cuda-version }}
+          bash .github/workflows/cuda/${{ runner.os }}.sh ${{ matrix.os }} ${{ matrix.cuda-version }}
 
       - name: Setup CUDA
         shell: bash