diff --git a/tests/experimental/weight_update/test_wu_controller.py b/tests/experimental/weight_update/test_wu_controller.py
index 302d9356b7..85c69a8dac 100644
--- a/tests/experimental/weight_update/test_wu_controller.py
+++ b/tests/experimental/weight_update/test_wu_controller.py
@@ -81,6 +81,7 @@ def test_connect_sends_correct_request(self, ctrl):
                 "save_path": "",
                 "use_lora": False,
                 "lora_name": "",
+                "colocate": False,
             },
             timeout=10.0,
         )
@@ -110,6 +111,7 @@ def test_connect_disk_mode_sends_disk_fields(self, ctrl):
                 "save_path": "/shared/weights",
                 "use_lora": True,
                 "lora_name": "my-lora",
+                "colocate": False,
             },
             timeout=10.0,
         )