diff --git a/pkgs/development/python-modules/ax/default.nix b/pkgs/development/python-modules/ax/default.nix
index dddcc6e1ebeb..7b3cc8f2b1c2 100644
--- a/pkgs/development/python-modules/ax/default.nix
+++ b/pkgs/development/python-modules/ax/default.nix
@@ -49,6 +49,8 @@ buildPythonPackage rec {
     pyre-extensions
   ];
 
+  env.ALLOW_BOTORCH_LATEST = "1";
+
   checkInputs = [
     hypothesis
     mercurial