diff --git a/.github/workflows/e2e_test.yml b/.github/workflows/e2e_test.yml
index 434f7a213..058ee593f 100644
--- a/.github/workflows/e2e_test.yml
+++ b/.github/workflows/e2e_test.yml
@@ -48,7 +48,7 @@ jobs:
         run: while [ $((curl localhost:7474/ > /dev/null 2>&1); echo $?) -ne 0 ]; do sleep 1; done
 
       - name: Setup test rails app
-        run: ./e2e_tests/setup.sh
+        run: ./e2e_tests/setup.sh ${{ github.sha }}
 
       - name: Install dependencies
         run: bundle update
diff --git a/e2e_tests/setup.sh b/e2e_tests/setup.sh
index e5f623b65..b69279fcf 100755
--- a/e2e_tests/setup.sh
+++ b/e2e_tests/setup.sh
@@ -7,8 +7,8 @@ if [[ -n "$ACTIVEGRAPH_PATH" ]]
 then
   sed 's|.*gem '"'"'activegraph'"'"'.*|gem '"'"'activegraph'"'"', path: "'"$ACTIVEGRAPH_PATH"'"|' docs/activegraph.rb > template.tmp
 else
-  echo "SHA=$(git rev-parse "$GITHUB_SHA")" >> $GITHUB_ENV
-  sed 's/.*gem '"'"'activegraph'"'"'.*/gem '"'"'activegraph'"'"', github: "neo4jrb\/activegraph", ref: "'"$(git rev-parse "$GITHUB_SHA")"'"/' docs/activegraph.rb > template.tmp
+  # echo "SHA=$(git rev-parse "$GITHUB_SHA")" >> $GITHUB_OUTPUT
+  sed 's/.*gem '"'"'activegraph'"'"'.*/gem '"'"'activegraph'"'"', github: "neo4jrb\/activegraph", ref: "'"$1"'"/' docs/activegraph.rb > template.tmp
 fi
 
 rails \_$ACTIVE_MODEL_VERSION\_ new myapp -O -m ./template.tmp