diff --git a/docs/source/conf.py b/docs/source/conf.py
index 51caaa9e..f73e286c 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -37,7 +37,7 @@
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
-extensions = ["sphinx.ext.todo", "sphinx.ext.viewcode", "sphinx.ext.napoleon", "myst_parser", "nbsphinx"]
+extensions = ["sphinx.ext.todo", "sphinx.ext.viewcode", "sphinx.ext.napoleon", "myst_parser", "myst_nb"]
 
 # Add any paths that contain templates here, relative tolibpython3.9.so.1.0: cannot open shared object file: No such file or this directory.
 templates_path = ['_templates']
diff --git a/requirements.txt b/requirements.txt
index 1d91ce49..36b1b012 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,4 +8,5 @@ sympy
 plotly
 scons
 myst_parser
+myst_nb
 portpicker
\ No newline at end of file