diff --git a/docs/requirements-docs.txt b/docs/requirements-docs.txt
index 3f9b956..8e47875 100644
--- a/docs/requirements-docs.txt
+++ b/docs/requirements-docs.txt
@@ -1,4 +1,11 @@
--r ../requirements-dev.txt
+elasticsearch>=7.7
+pandas>=1.2.0
+matplotlib
+nbval
+numpydoc>=0.9.0
+scikit-learn>=0.22.1
+xgboost>=1
+lightgbm
 nbsphinx
 numpydoc>=0.9.0
 git+https://github.com/pandas-dev/pydata-sphinx-theme.git@master
diff --git a/tests/setup_tests.py b/tests/setup_tests.py
index dcea39c..228e423 100644
--- a/tests/setup_tests.py
+++ b/tests/setup_tests.py
@@ -90,7 +90,12 @@ def _update_max_compilations_limit(es, limit="10000/1m"):
     if es_version(es) < (7, 8):
         body = {"transient": {"script.max_compilations_rate": limit}}
     else:
-        body = {"transient": {"script.context.field.max_compilations_rate": limit}}
+        body = {
+            "transient": {
+                "script.max_compilations_rate": "use-context",
+                "script.context.field.max_compilations_rate": limit,
+            }
+        }
     es.cluster.put_settings(body=body)