diff --git a/setup.py b/setup.py
index c89937bb4..0f349464a 100644
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,7 @@
     "pandas",
     "aepsych_client==0.3.0",
     "statsmodels",
-    "ax-platform==0.3.5",
+    "ax-platform==0.3.6",
 ]
 
 BENCHMARK_REQUIRES = ["tqdm", "pathos", "multiprocess"]