diff --git a/Dockerfile b/Dockerfile
index d700b53608abcf80852a5abfc4c830cf659d3909..3dad99ec083bc72ebbff7a9311182543dcf516b8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -39,7 +39,8 @@ RUN pip install pandaSDMX
 RUN pip install plotly
 
 # Install for -> artifint
-RUN pip install tensorflow_decision_forests
+RUN pip install tensorflow
+# RUN pip install tensorflow_decision_forests
 RUN pip install ortools
 RUN pip install python-constraint