diff --git a/template/before.sh.erb b/template/before.sh.erb
index 6b89042..215a22e 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 718951c..09c2e6b 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 %>