diff --git a/scripts/ci/start-api-benchmark.sh b/scripts/ci/start-api-benchmark.sh
index 0cf83c83c27664a5bc0ae386630401d4873af051..fa6a50129fa8b290315cd83dc5a47b8708214dca 100755
--- a/scripts/ci/start-api-benchmark.sh
+++ b/scripts/ci/start-api-benchmark.sh
@@ -18,6 +18,7 @@ echo "Script base dir is: $BASE_DIR"
 
 pip install tox
 pip install prettytable
+pip install pyyaml
 
 echo "Creating benchmark test file as: $4.py"
 $BASE_DIR/tests/tests_api/hivemind/benchmarks/benchmark_generator.py $3 "$4.py" "http://$1:$2"