diff --git a/.devcontainer/ros2-feature-smoke/devcontainer.json b/.devcontainer/ros2-feature-smoke/devcontainer.json
index 52bb264..7cbd7ad 100644
--- a/.devcontainer/ros2-feature-smoke/devcontainer.json
+++ b/.devcontainer/ros2-feature-smoke/devcontainer.json
@@ -2,7 +2,9 @@
   "name": "ros2-feature-smoke",
   "image": "mcr.microsoft.com/devcontainers/base:ubuntu-24.04",
   "features": {
-    "../local-features/ros2": {}
+    "../local-features/ros2": {
+      "package": "ros-core"
+    }
   },
-  "postCreateCommand": "bash -lc 'ros2 pkg prefix rviz2 >/dev/null && command -v colcon >/dev/null && command -v rosdep >/dev/null && command -v vcs >/dev/null'"
+  "postCreateCommand": "bash -lc 'command -v colcon >/dev/null && command -v rosdep >/dev/null && command -v vcs >/dev/null'"
 }
diff --git a/features/src/ros2/install.sh b/features/src/ros2/install.sh
index 034ef91..2350740 100755
--- a/features/src/ros2/install.sh
+++ b/features/src/ros2/install.sh
@@ -178,6 +178,4 @@ else
     echo "No effective Dev Container user was provided; skipping user-specific shell configuration."
 fi
 
-rm -rf /var/lib/apt/lists/*
-
 echo "Configured ROS 2 ${ros_distro} (${ros_package}) development environment."