diff --git a/.travis.yml b/.travis.yml
index 476ba445..048f5c7a 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -21,12 +21,12 @@ install:
 - conda config --set always_yes yes --set changeps1 no
 - conda update -q conda
 - conda info -a
-- conda install -c conda-forge nodejs notebook matplotlib ipywidgets selenium jupyterlab=1
+- conda install -c conda-forge nodejs notebook matplotlib ipywidgets jupyterlab
 - pip install -v .
 script:
 - jupyter nbextension install --py --symlink --sys-prefix ipympl
 - jupyter nbextension enable --py --sys-prefix ipympl
 - jupyter labextension list 2>&1 | grep -q jupyter-matplotlib
 - jupyter labextension install @jupyter-widgets/jupyterlab-manager
-- python -m jupyterlab.selenium_check --NotebookApp.ip=127.0.0.1
 - jupyter labextension link js
+- python -m jupyterlab.browser_check --NotebookApp.ip=127.0.0.1
diff --git a/js/package.json b/js/package.json
index 9521d7f7..ba7b7154 100644
--- a/js/package.json
+++ b/js/package.json
@@ -30,6 +30,6 @@
     "webpack-cli": "^3.3.11"
   },
   "dependencies": {
-    "@jupyter-widgets/base": "^2.0.0"
+    "@jupyter-widgets/base": "^2 || ^3"
   }
 }