diff --git a/setup.py b/setup.py
index ec71745..aeb087b 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name="miplearn",
-    version="0.2.0.dev1",
+    version="0.2.0.dev2",
     author="Alinson S. Xavier",
     author_email="axavier@anl.gov",
     description="Extensible framework for Learning-Enhanced Mixed-Integer Optimization",
@@ -29,6 +29,7 @@ setup(
         "python-markdown-math",
         "seaborn",
         "sklearn",
+        "overrides",
         "tqdm",
     ],
 )