diff --git a/templates/ros/docker/Dockerfile b/templates/ros/docker/Dockerfile
index f2ae50c..7077dbd 100644
--- a/templates/ros/docker/Dockerfile
+++ b/templates/ros/docker/Dockerfile
@@ -31,6 +31,7 @@ ENV DEBIAN_FRONTEND=dialog
 #####################
 FROM base as dev
 
+ARG CATKIN_WORKSPACE_DIR="/catkin_ws"
 ARG USERNAME="developer"
 ARG UID=1000
 ARG GID=1000
diff --git a/templates/ros2/docker/Dockerfile b/templates/ros2/docker/Dockerfile
index 17624f6..120938f 100644
--- a/templates/ros2/docker/Dockerfile
+++ b/templates/ros2/docker/Dockerfile
@@ -36,6 +36,7 @@ ENV DEBIAN_FRONTEND=dialog
 #####################
 FROM base as dev
 
+ARG AMENT_WORKSPACE_DIR="/ament_ws"
 ARG USERNAME="developer"
 ARG UID=1000
 ARG GID=1000