diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 08783e0a1e0..ee17d2800b0 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -52,5 +52,4 @@ RUN git clone https://github.com/ROCmSoftwarePlatform/rocmProfileData.git \
 
 COPY docker/libamdhip64.so.6 /opt/rocm/lib/libamdhip64.so.6.0.60000
 COPY docker/libfile_plugin.so /opt/rocm/lib/roctracer
-COPY docker/run_13b.sh $WORKSPACE_DIR/
-COPY docker/run_70b.sh $WORKSPACE_DIR/
+ENV WORKSPACE_DIR=/workspace/vllm