diff --git a/Dockerfile b/Dockerfile
index de522c80052c1b89960093b2ec665c0396ebfc13..142eaf6ad8adc4d6137e3df76fc77b9b02031b6a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -35,9 +35,9 @@ RUN pip install plotly
 
 # Install for -> artifint
 RUN pip install python-constraint
+RUN pip install tensorflow_decision_forests
 RUN pip install protobuf==3.19
 RUN pip install ortools==9.0.9048
-#RUN pip install tensorflow_decision_forests
 
 # Install for -> windkanl
 RUN pip install paho-mqtt