diff --git a/ax/modelbridge/tests/test_generation_node.py b/ax/modelbridge/tests/test_generation_node.py
index 67b96316ccc..17d4e32d67f 100644
--- a/ax/modelbridge/tests/test_generation_node.py
+++ b/ax/modelbridge/tests/test_generation_node.py
@@ -313,7 +313,7 @@ def setUp(self) -> None:
         ms_gpei = ModelSpec(model_enum=Models.GPEI)
         ms_gpei.fit(experiment=self.branin_experiment, data=data)
 
-        ms_botorch = ModelSpec(model_enum=Models.BOTORCH)
+        ms_botorch = ModelSpec(model_enum=Models.BOTORCH_MODULAR)
         ms_botorch.fit(experiment=self.branin_experiment, data=data)
 
         self.fitted_model_specs = [ms_gpei, ms_botorch]