diff --git a/ocdskingfishercolab/__init__.py b/ocdskingfishercolab/__init__.py
index abac52c..8f64e30 100644
--- a/ocdskingfishercolab/__init__.py
+++ b/ocdskingfishercolab/__init__.py
@@ -16,7 +16,7 @@
 from gspread_dataframe import set_with_dataframe
 from IPython import get_ipython
 from IPython.display import HTML
-from notebook import notebookapp
+from jupyter_server import serverapp
 from oauth2client.client import GoogleCredentials
 from oauth2client.contrib.gce import AppAssertionCredentials
 from pydrive2.auth import GoogleAuth
@@ -338,7 +338,7 @@ def write_data_as_json(data, filename):
 
 
 def _notebook_id():
-    server = next(notebookapp.list_running_servers())
+    server = next(serverapp.list_running_servers())
     return requests.get(urljoin(server['url'], 'api/sessions')).json()[0]['path'][7:]  # fileId=
 
 
diff --git a/setup.cfg b/setup.cfg
index d171dce..5244f7b 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -25,8 +25,8 @@ install_requires =
     httplib2
     ipython
     ipython-sql~=0.4.0
+    jupyter_server
     matplotlib
-    notebook
     oauth2client
     pydrive2
     requests