diff --git a/template/before.sh.erb b/template/before.sh.erb
index 6b89042428b7fb2e004cdb235f8d85cd18385ab1..215a22ecb1ea248fff982ad7b08c17c94d8c1b4d 100755
--- a/template/before.sh.erb
+++ b/template/before.sh.erb
@@ -54,5 +54,6 @@ c.NotebookApp.open_browser = False
 c.NotebookApp.allow_origin = '*'
 c.NotebookApp.notebook_dir = '${HOME}'
 c.NotebookApp.disable_check_xsrf = True
+c.NotebookApp.nbserver_extensions = {"jupyterlmod": True}
 EOL
 )
diff --git a/template/script.sh.erb b/template/script.sh.erb
index 718951c8076fdf1f2b30da164f99f6b5d8a87672..09c2e6bed738ea3f8fd2583fbe3bd5953ce4c979 100755
--- a/template/script.sh.erb
+++ b/template/script.sh.erb
@@ -26,4 +26,4 @@ echo "TIMING - Starting jupyter at: $(date)"
 
 # Launch the Jupyter Notebook Server
 set -x
-jupyter notebook --config="${CONFIG_FILE}" <%= context.extra_jupyter_args %>
+PYTHONPATH=$HOME/.local/lib/python3.6/site-packages jupyter notebook --config="${CONFIG_FILE}" <%= context.extra_jupyter_args %>