diff --git a/.github/workflows/dockerization.yml b/.github/workflows/dockerization.yml
index a69d6953945..dda71488490 100644
--- a/.github/workflows/dockerization.yml
+++ b/.github/workflows/dockerization.yml
@@ -29,7 +29,7 @@ jobs:
         fx workspace create --prefix example_workspace --template keras_cnn_mnist
         cd example_workspace
         fx plan initialize -a localhost
-        fx workspace dockerize --save --revision https://github.com/securefederatedai/openfl.git@develop
+        fx workspace dockerize --save --revision https://github.com/${GITHUB_REPOSITORY}.git@${GITHUB_HEAD_REF}
 
     - name: Create certificate authority for workspace
       run: |