diff --git a/lib/src/cli/templates/test_optimizer_bundle.dart b/lib/src/cli/templates/test_optimizer_bundle.dart
index 4163b8b7e..f521b93ca 100644
--- a/lib/src/cli/templates/test_optimizer_bundle.dart
+++ b/lib/src/cli/templates/test_optimizer_bundle.dart
@@ -35,7 +35,7 @@ final testOptimizerBundle = MasonBundle.fromJson(<String, dynamic>{
     {
       "path": "pubspec.yaml",
       "data":
-          "bmFtZTogaG9va3MKcHVibGlzaF90bzogbm9uZQoKZW52aXJvbm1lbnQ6CiAgc2RrOiBeMy44LjAKCmRlcGVuZGVuY2llczoKICBtYXNvbjogXjAuMS4wCiAgcGF0aDogXjEuOC4xCgojIEJld2FyZTogb24gaG9va3MsIGV2ZW4gZGV2IGRlcGVuZGVuY2llcyBoYXZlIHRvIGJlIGNvbXBhdGlibGUgdG8gYWxsIGRhcnQgdmVyc2lvbnMgY292ZXJlZCBieQojIHRoZSBzZGsgY29uc3RyYWludHMgYWJvdmUKZGV2X2RlcGVuZGVuY2llczoKICBtb2NrdGFpbDogXjEuMC4wCiAgdGVzdDogXjEuMjUuMAogIHZlcnlfZ29vZF9hbmFseXNpczogXjkuMC4wCg==",
+          "bmFtZTogaG9va3MKcHVibGlzaF90bzogbm9uZQoKZW52aXJvbm1lbnQ6CiAgc2RrOiBeMy45LjAKCmRlcGVuZGVuY2llczoKICBtYXNvbjogXjAuMS4wCiAgcGF0aDogXjEuOC4xCgojIEJld2FyZTogb24gaG9va3MsIGV2ZW4gZGV2IGRlcGVuZGVuY2llcyBoYXZlIHRvIGJlIGNvbXBhdGlibGUgdG8gYWxsIGRhcnQgdmVyc2lvbnMgY292ZXJlZCBieQojIHRoZSBzZGsgY29uc3RyYWludHMgYWJvdmUKZGV2X2RlcGVuZGVuY2llczoKICBtb2NrdGFpbDogXjEuMC4wCiAgdGVzdDogXjEuMjUuMAogIHZlcnlfZ29vZF9hbmFseXNpczogXjkuMC4wCg==",
       "type": "text",
     },
     {
diff --git a/tool/generate_test_optimizer_bundle.sh b/tool/generate_test_optimizer_bundle.sh
index 89ca6b8cb..74cb754be 100755
--- a/tool/generate_test_optimizer_bundle.sh
+++ b/tool/generate_test_optimizer_bundle.sh
@@ -1,8 +1,8 @@
 #!/bin/bash
-BUNDLE_OUTPUT_DIR="lib/src/commands/test/templates"
+BUNDLE_OUTPUT_DIR="lib/src/cli/templates"
 mason bundle --source path ./bricks/test_optimizer/ -t dart --output-dir $BUNDLE_OUTPUT_DIR
 
-input="lib/src/commands/test/templates/test_optimizer_bundle.dart"
+input="lib/src/cli/templates/test_optimizer_bundle.dart"
 echo -e "// To generate this file, run: tool/generate_test_optimizer_bundle.sh\n$(cat $input)" > $input
 
 dart format $BUNDLE_OUTPUT_DIR
\ No newline at end of file