diff --git a/requirements.txt b/requirements.txt
index 105cebf..14c9701 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 pybind11>=2.4
-infomap==1.0.6
+infomap>=1.0.6,<3
 folium>=0.7.0
 numpy
 tqdm
-scikit-learn
\ No newline at end of file
+scikit-learn
diff --git a/setup.py b/setup.py
index 5d85115..2e3a58f 100644
--- a/setup.py
+++ b/setup.py
@@ -114,7 +114,7 @@ def build_extensions(self):
     setup_requires=["pybind11>=2.4"],
     install_requires=[
         "pybind11>=2.4",
-        "infomap==1.0.6",
+        "infomap>=1.0.6,<3",
         "folium>=0.7.0",
         "numpy",
         "tqdm",