diff --git a/mkdocs.yml b/mkdocs.yml
index ac029af9..91831b3a 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -56,6 +56,7 @@ plugins:
       binder: true
       binder_service_name: "gh"
       binder_branch: "master"
+      binder_ui: "lab"
 
 # Taken from here: https://squidfunk.github.io/mkdocs-material/extensions/codehilite/
 markdown_extensions:
diff --git a/requirements.txt b/requirements.txt
index 9710215f..cf9bee95 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ jupyter==1.0.0
 matplotlib==3.3.4
 mkdocs-material==7.0.6
 mkdocs==1.1.2
-mknotebooks==0.6.2
+git+https://github.com/greenape/mknotebooks.git
 nbconvert==6.0.7
 networkx==2.5
 numpy==1.20.1
diff --git a/scripts/ci/build_environment.sh b/scripts/ci/build_environment.sh
index 264c85ec..e0bbebf8 100644
--- a/scripts/ci/build_environment.sh
+++ b/scripts/ci/build_environment.sh
@@ -2,7 +2,7 @@
 conda install -c conda-forge mamba
 conda env create -f environment.yml
 conda activate nams || source activate nams
-python -m pip install --no-deps mknotebooks
+python -m pip install --no-deps git+https://github.com/greenape/mknotebooks.git
 python -m pip install --no-deps .
 python -m pip install .
 conda install conda-pack