diff --git a/requirements/docs.txt b/requirements/docs.txt
index 6f335f911f..0092cf5ca8 100644
--- a/requirements/docs.txt
+++ b/requirements/docs.txt
@@ -1,7 +1,9 @@
 attrs==21.4.0
 Jinja2<3.1
 linkify-it-py==1.0.3
-myst-nb
+myst-nb==0.17.2
+nbformat==5.9.2
+jsonschema==4.17.3
 natsort==8.1.0
 recommonmark==0.7.1
 sphinx_book_theme~=1.0.1
diff --git a/tox.ini b/tox.ini
index e35a7dcc78..a6c20fa764 100644
--- a/tox.ini
+++ b/tox.ini
@@ -121,7 +121,7 @@ commands =
 changedir = {toxinidir}
 deps =
     -rrequirements/docs.txt
-    -rrequirements/test.txt
+    -rrequirements/dev.txt
     git+https://github.com/NVIDIA-Merlin/core.git
     git+https://github.com/NVIDIA-Merlin/dataloader.git
     git+https://github.com/NVIDIA-Merlin/NVTabular.git