diff --git a/scripts/import-tracing-runtime.sh b/scripts/import-tracing-runtime.sh
index 47d1169..5ff0a8d 100755
--- a/scripts/import-tracing-runtime.sh
+++ b/scripts/import-tracing-runtime.sh
@@ -2,6 +2,7 @@
 
 SPEC_VERSION=$1
 GIT_REF=${2:-"runtime-$SPEC_VERSION"}
+GIT_REPO=${3:-"https://github.com/moonbeam-foundation/moonbeam"}
 
 if [[ "$GIT_REF" == "runtime-$SPEC_VERSION" ]]; then
   GIT_DEP_REF="--rev $GIT_REF"
@@ -18,7 +19,7 @@ else
   echo "Get moonbeam snapshot..."
   rm -rf tmp
   mkdir tmp
-  git clone https://github.com/moonbeam-foundation/moonbeam --depth 1 -b $GIT_REF $MOONBEAM_PATH
+  git clone $GIT_REPO --depth 1 -b $GIT_REF $MOONBEAM_PATH
 fi
 
 # Copy relevant files
@@ -35,7 +36,7 @@ rm -rf tracing/$SPEC_VERSION/runtime/relay-encoder
 
 echo "Run migration script"
 cd scripts
-cargo run -q --bin migrate-imported -- --dir ../tracing/$SPEC_VERSION --repo "https://github.com/moonbeam-foundation/moonbeam" $GIT_DEP_REF
+cargo run -q --bin migrate-imported -- --dir ../tracing/$SPEC_VERSION --repo "$GIT_REPO" $GIT_DEP_REF
 cd ..
 
 echo "Running ./scripts/update-tracing-runtime-lock.sh $SPEC_VERSION ..."