diff --git a/.gitmodules b/.gitmodules
index 0162c3dc7cf..8bc3af6a4a0 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,4 +1,4 @@
 [submodule "lib/stan_math"]
 	path = lib/stan_math
 	url = https://github.com/stan-dev/math.git
-        branch = mpi_warmup_v2
+        branch = mpi_warmup_framework