diff --git a/Dockerfile.hardware b/Dockerfile.hardware
index e6533ac..75916a5 100644
--- a/Dockerfile.hardware
+++ b/Dockerfile.hardware
@@ -15,7 +15,9 @@ RUN apt-get update  && \
     # Setup workspace
     git clone -b ros2 https://github.com/husarion/panther_ros.git src/panther_ros && \
     vcs import src < src/panther_ros/panther/panther_hardware.repos && \
-    cp -r src/ros2_controllers/imu_sensor_broadcaster src && rm -rf src/ros2_controllers && \
+    cp -r src/ros2_controllers/diff_drive_controller src && \
+    cp -r src/ros2_controllers/imu_sensor_broadcaster src && \
+    rm -rf src/ros2_controllers && \
     # Install dependencies
     rosdep init && \
     rosdep update --rosdistro $ROS_DISTRO && \