diff --git a/mlw_packages.yml b/mlw_packages.yml
index 5efe756929a403ce12ff3473712a8d1da38b8d2c..a317360c79119228f4eabb0552e869b83290941c 100644
--- a/mlw_packages.yml
+++ b/mlw_packages.yml
@@ -17,5 +17,4 @@ dependencies:
     - seaborn
     - jupyter
     - keras
-    - graphviz
-    - python-graphviz
+    - tensorflow==1.10