diff --git a/integration-tests/scripts/download-polkadot.sh b/integration-tests/scripts/download-polkadot.sh
index cb57238f3..4b0079588 100755
--- a/integration-tests/scripts/download-polkadot.sh
+++ b/integration-tests/scripts/download-polkadot.sh
@@ -9,8 +9,9 @@ if [[ $(uname) == "Darwin" ]]; then
     exit 1
 fi
 
-# Version 1.4.0 of relaychain didn't allow the parachain to produce blocks
-# that's why we use 1.1.0
+# TODO: Use following line instead of the one above once Zeitgeist uses >=polkadot-v1.1.0
+# Note: Version 1.4.0 of relaychain didn't allow the parachain to produce blocks
+# polkadot_release=$(echo $branch | sed 's/#.*//' | sed 's/\/polkadot-sdk?branch=polkadot-v//' | sed 's/-.*//')
 polkadot_release="1.1.0"
 
 # Always run the commands from the "test" dir