diff --git a/docker/federation/run-federation-test.bash b/docker/federation/run-federation-test.bash
index bc73fff63..77cc981f4 100755
--- a/docker/federation/run-federation-test.bash
+++ b/docker/federation/run-federation-test.bash
@@ -6,7 +6,7 @@ pushd ../../server/ || exit
 cargo build &
 popd || exit
 
-if [ "$1" = "-yarn" ]; then
+if [ "$1" != "--no-yarn-build" ]; then
   pushd ../../ui/ || exit
   yarn
   yarn build