diff --git a/suites/upgrade/firefly-x/parallel/4-final-workload/rados_mon_thrash.yaml b/suites/upgrade/firefly-x/parallel/4-final-workload/rados_mon_thrash.yaml
index f1e30f241..9f1867937 100644
--- a/suites/upgrade/firefly-x/parallel/4-final-workload/rados_mon_thrash.yaml
+++ b/suites/upgrade/firefly-x/parallel/4-final-workload/rados_mon_thrash.yaml
@@ -4,7 +4,8 @@ tasks:
       thrash_delay: 1
   - print: "**** done mon_thrash 4-final-workload"
   - workunit:
+      branch: firefly
       clients:
         client.1:
-          - rados/test.sh
-  - print: "**** done rados/test.sh 4-final-workload"
+          - rados/test-upgrade-v9.0.1.sh
+  - print: "**** done rados/test-upgrade-v9.0.1.sh 4-final-workload"
diff --git a/suites/upgrade/giant-x/parallel/4-final-workload/rados_mon_thrash.yaml b/suites/upgrade/giant-x/parallel/4-final-workload/rados_mon_thrash.yaml
index f1e30f241..fcd6963da 100644
--- a/suites/upgrade/giant-x/parallel/4-final-workload/rados_mon_thrash.yaml
+++ b/suites/upgrade/giant-x/parallel/4-final-workload/rados_mon_thrash.yaml
@@ -4,7 +4,8 @@ tasks:
       thrash_delay: 1
   - print: "**** done mon_thrash 4-final-workload"
   - workunit:
+      branch: giant
       clients:
         client.1:
-          - rados/test.sh
-  - print: "**** done rados/test.sh 4-final-workload"
+          - rados/test-upgrade-v9.0.1.sh
+  - print: "**** done rados/test-upgrade-v9.0.1.sh 4-final-workload"
diff --git a/suites/upgrade/hammer-x/parallel/4-final-workload/rados_mon_thrash.yaml b/suites/upgrade/hammer-x/parallel/4-final-workload/rados_mon_thrash.yaml
index f1e30f241..50860783f 100644
--- a/suites/upgrade/hammer-x/parallel/4-final-workload/rados_mon_thrash.yaml
+++ b/suites/upgrade/hammer-x/parallel/4-final-workload/rados_mon_thrash.yaml
@@ -4,7 +4,8 @@ tasks:
       thrash_delay: 1
   - print: "**** done mon_thrash 4-final-workload"
   - workunit:
+      branch: hammer
       clients:
         client.1:
-          - rados/test.sh
-  - print: "**** done rados/test.sh 4-final-workload"
+          - rados/test-upgrade-v9.0.1.sh
+  - print: "**** done rados/test-upgrade-v9.0.1.sh 4-final-workload"