diff --git a/setup.py b/setup.py
index 9951715..7bcef7f 100644
--- a/setup.py
+++ b/setup.py
@@ -2,14 +2,7 @@
 
 extras_require = {
     'develop': {'pytest', 'flake8>=3.9.0', 'black'},
-    'local': [
-        'pydotplus==2.0.2',
-        'adage>=0.10.1',
-        'packtivity>=0.14.23',
-        'yadage>=0.20.1',  # yadage[viz] breaks so install following manually
-        'pydot',  # from yadage[viz] extra
-        'pygraphviz',  # from yadage[viz] extra
-    ],
+    'local': ['packtivity>=0.14.23', 'yadage[viz]>=0.20.2'],
     'kubernetes': ['kubernetes==9.0.0'],
     'reana': ['reana-client>=0.8.0'],
 }