diff --git a/contract/scripts/run-chain.sh b/contract/scripts/run-chain.sh
index 689fc181..019cfa9f 100755
--- a/contract/scripts/run-chain.sh
+++ b/contract/scripts/run-chain.sh
@@ -1,10 +1,12 @@
 #!/bin/bash
 
+. /usr/src/upgrade-test-scripts/env_setup.sh
+
 # Start the chain in the background
 /usr/src/upgrade-test-scripts/start_agd.sh &
 
 # wait for blocks to start being produced
-waitForBlock 2
+waitForBlock 1
 
 make -C /workspace/contract mint100