diff --git a/.github/workflows/humble-docker-build.yaml b/.github/workflows/humble-docker-build.yaml
index d9130d3e..ad9b2b57 100644
--- a/.github/workflows/humble-docker-build.yaml
+++ b/.github/workflows/humble-docker-build.yaml
@@ -18,5 +18,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v4
+      with:
+        ref: humble
     - name: Build the Docker image
-      run: docker build . --file Dockerfile/Dockerfile --tag ros2_control_demos_humble
+      run: docker build --file Dockerfile/Dockerfile --tag ros2_control_demos_humble --build-arg ROS_DISTRO=humble .
diff --git a/.github/workflows/iron-docker-build.yaml b/.github/workflows/iron-docker-build.yaml
index fb95d212..c1f89dc4 100644
--- a/.github/workflows/iron-docker-build.yaml
+++ b/.github/workflows/iron-docker-build.yaml
@@ -18,5 +18,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v4
+      with:
+        ref: iron
     - name: Build the Docker image
-      run: docker build . --file Dockerfile/Dockerfile --tag ros2_control_demos_iron
+      run: docker build --file Dockerfile/Dockerfile --tag ros2_control_demos_iron --build-arg ROS_DISTRO=iron .
diff --git a/.github/workflows/rolling-docker-build.yaml b/.github/workflows/rolling-docker-build.yaml
index cfb8b091..02e234a6 100644
--- a/.github/workflows/rolling-docker-build.yaml
+++ b/.github/workflows/rolling-docker-build.yaml
@@ -18,5 +18,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v4
+      with:
+        ref: master
     - name: Build the Docker image
-      run: docker build . --file Dockerfile/Dockerfile --tag ros2_control_demos_rolling
+      run: docker build --file Dockerfile/Dockerfile --tag ros2_control_demos_rolling --build-arg ROS_DISTRO=rolling .