diff --git a/Dockerfile b/Dockerfile
index a77bd68..54c00ce 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -23,6 +23,7 @@ RUN pip install -r requirements.txt
 # install required ros dependencies
 RUN apt-get install -y ros-$ROS_DISTRO-cv-bridge
 RUN apt-get install -y ros-$ROS_DISTRO-pcl-ros
+RUN apt-get install -y ros-$ROS_DISTRO-image-proc
 
 # socket io
 RUN apt-get install -y netbase
@@ -30,3 +31,4 @@ RUN apt-get install -y netbase
 RUN mkdir /capstone
 VOLUME ["/capstone"]
 VOLUME ["/root/.ros/log/"]
+WORKDIR /capstone/ros
diff --git a/requirements.txt b/requirements.txt
index 4ad5cdb..8522d13 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,5 +6,5 @@ numpy>=1.13.1
 Pillow>=2.2.1
 scipy
 keras
-tensorflow=1.0.0
+tensorflow==1.0.0
 h5py