diff --git a/README.md b/README.md
index f76d8bb..ee72d4f 100644
--- a/README.md
+++ b/README.md
@@ -391,15 +391,17 @@ Then, use it as seen at the bottom of the
 [What_If_Tool_Notebook_Usage.ipynb notebook](https://colab.research.google.com/github/pair-code/what-if-tool/blob/master/What_If_Tool_Notebook_Usage.ipynb).
 
 ### How do I enable it for use in a JupyterLab or Cloud AI Platform notebook?
-WIT has been tested in JupyterLab versions 1.x and 2.x.
+WIT has been tested in JupyterLab versions 1.x, 2.x, and 3.x.
 
-Install and enable WIT for JupyterLab 2.x by running a cell containing:
+Install and enable WIT for JupyterLab 3.x by running a cell containing:
 ```
 !pip install witwidget
 !jupyter labextension install wit-widget
-!jupyter labextension install @jupyter-widgets/jupyterlab-manager@2
+!jupyter labextension install @jupyter-widgets/jupyterlab-manager
 ```
-Note that you need to specify the correct version of jupyterlab-manager for your JupyterLab version as per https://www.npmjs.com/package/@jupyter-widgets/jupyterlab-manager.
+Note that you may need to specify the correct version of jupyterlab-manager for
+you JupyterLab version as per
+https://www.npmjs.com/package/@jupyter-widgets/jupyterlab-manager.
 
 Note that you may need to run `!sudo jupyter labextension ...` commands depending on your notebook setup.
 
diff --git a/witwidget/notebook/jupyter/js/package.json b/witwidget/notebook/jupyter/js/package.json
index c9123e2..91adf1b 100644
--- a/witwidget/notebook/jupyter/js/package.json
+++ b/witwidget/notebook/jupyter/js/package.json
@@ -3,7 +3,7 @@
   "description": "What-If Tool jupyter widget",
   "author": "Google LLC",
   "main": "dist/index.js",
-  "version": "1.8.0",
+  "version": "1.8.1",
   "license" : "Apache 2.0",
   "homepage": "https://github.com/pair-code/what-if-tool",
   "keywords": [
@@ -26,7 +26,7 @@
     "webpack": "^3.5.5"
   },
   "dependencies": {
-    "@jupyter-widgets/base": "^1.1 || ^2 || ^3"
+    "@jupyter-widgets/base": "^1.1 || ^2 || ^3 || ^4"
   },
   "jupyterlab": {
     "extension": "lib/labplugin"
diff --git a/witwidget/pip_package/setup.py b/witwidget/pip_package/setup.py
index 1064914..014aec9 100644
--- a/witwidget/pip_package/setup.py
+++ b/witwidget/pip_package/setup.py
@@ -37,7 +37,6 @@
     'absl-py >= 0.4',
     'google-api-python-client>=1.7.8',
     'ipywidgets>=7.0.0',
-    'jupyter>=1.0,<2',
     'oauth2client>=4.1.3',
     'six>=1.12.0',
 ] + _TF_REQ
diff --git a/witwidget/version.py b/witwidget/version.py
index 2f387b1..241c930 100644
--- a/witwidget/version.py
+++ b/witwidget/version.py
@@ -14,4 +14,4 @@
 
 """Contains the version string."""
 
-VERSION = '1.8.0'
+VERSION = '1.8.1'