diff --git a/beakerx_tabledisplay/beakerx_tabledisplay/install.py b/beakerx_tabledisplay/beakerx_tabledisplay/install.py
index 237dfde..3f49f86 100644
--- a/beakerx_tabledisplay/beakerx_tabledisplay/install.py
+++ b/beakerx_tabledisplay/beakerx_tabledisplay/install.py
@@ -25,7 +25,7 @@
 
 def install(args):
     subprocess.check_call(["jupyter", "serverextension", "enable", "beakerx_tabledisplay", "--py", "--sys-prefix"])
-    if LAB_VERSION is not None and LAB_VERSION != 3:
+    if LAB_VERSION is not None and LAB_VERSION < 3:
         subprocess.call(["jupyter", "labextension", "install", "@jupyter-widgets/jupyterlab-manager", "--no-build"],
                        stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
         if LAB_VERSION == 1:
@@ -38,7 +38,7 @@ def uninstall(args):
     subprocess.check_call(["jupyter", "nbextension", "disable", "beakerx_tabledisplay", "--py", "--sys-prefix"])
     subprocess.check_call(["jupyter", "nbextension", "uninstall", "beakerx_tabledisplay", "--py", "--sys-prefix"])
     subprocess.check_call(["jupyter", "serverextension", "disable", "beakerx_tabledisplay", "--py", "--sys-prefix"])
-    if LAB_VERSION is not None:
+    if LAB_VERSION is not None and LAB_VERSION < 3:
         subprocess.call(["jupyter", "labextension", "uninstall", "beakerx-jupyterlab"],
                         stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)