diff --git a/setup.py b/setup.py
index 531f02e4..76be3191 100644
--- a/setup.py
+++ b/setup.py
@@ -31,7 +31,7 @@ def test_suite():
     install_requires=[
         "numpy",
         "aiida-core",
-        "node-graph==0.0.1",
+        "node-graph",
         "cloudpickle",
         "aiida-pseudo",
         "aiida-quantumespresso",