diff --git a/environment.yml b/environment.yml
index 9fdcc80..4c0ebf8 100644
--- a/environment.yml
+++ b/environment.yml
@@ -15,6 +15,7 @@ dependencies:
   - scikit-image=0.14.2
   - jupyter=1.0.0
   - tqdm=4.55.2
+  - ipympl=0.9.3
   - pip
   - pip:
       - opencv-python==4.2.0.34
diff --git a/requirements.txt b/requirements.txt
index a0fda1b..c3c3225 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,6 +2,7 @@ numpy
 matplotlib
 scipy
 ipywidgets
+ipympl
 scikit-image
 jupyter
 opencv-python
diff --git a/setup.py b/setup.py
index 48d989b..85373fc 100644
--- a/setup.py
+++ b/setup.py
@@ -16,6 +16,7 @@
         "matplotlib",
         "scipy",
         "ipywidgets",
+        "ipympl",
         "scikit-image",
         "jupyter",
         "opencv-python",
@@ -24,7 +25,7 @@
         "tqdm",
         "pandas",
         "absl-py",
-        "matplotlib-polyroi==0.1.6",
+        "matplotlib-polyroi>=0.1.6",
     ],
     description="Cell cortex segmentation in C. elegans PAR protein images",
     long_description=long_description,