diff --git a/tests/tiup-playground/test_playground.sh b/tests/tiup-playground/test_playground.sh
index 98fab3b32b..01c93029e6 100755
--- a/tests/tiup-playground/test_playground.sh
+++ b/tests/tiup-playground/test_playground.sh
@@ -69,11 +69,9 @@ function wait_tiup() {
 	while ! grep -q "TiDB Playground Cluster is started" $outfile; do
 		sleep 1
 	done
-	while ! tiup-playground display; do
-		sleep 1
-	done
 }
-export -f tiup-playground wait_tiup
+export outfile
+export -f wait_tiup
 timeout 300s bash -c wait_tiup
 tiup-playground display
 tiup-playground scale-out --db 2