diff --git a/Dockerfile.ref b/Dockerfile.ref
index 6c9de2ff..906e2229 100644
--- a/Dockerfile.ref
+++ b/Dockerfile.ref
@@ -18,4 +18,7 @@ COPY ./vllm/core/block_manager_v2.py /workspace/qwen3_6_scripts/vendor_overrides
 COPY ./vllm/sampling_params.py /workspace/qwen3_6_scripts/vendor_overrides/vllm/sampling_params.py
 COPY ./vllm/model_executor/sampling_metadata.py /workspace/qwen3_6_scripts/vendor_overrides/vllm/model_executor/sampling_metadata.py
 COPY ./vllm/model_executor/layers/sampler.py /workspace/qwen3_6_scripts/vendor_overrides/vllm/model_executor/layers/sampler.py
-RUN cd ./qwen3_6_scripts && bash ./patch_ops.sh
+COPY ./vllm/spec_decode/spec_decode_worker.py /workspace/qwen3_6_scripts/vendor_overrides/vllm/spec_decode/spec_decode_worker.py
+COPY ./vllm/spec_decode/multi_step_worker.py /workspace/qwen3_6_scripts/vendor_overrides/vllm/spec_decode/multi_step_worker.py
+COPY ./vllm/spec_decode/metrics.py /workspace/qwen3_6_scripts/vendor_overrides/vllm/spec_decode/metrics.py
+RUN cd ./qwen3_6_scripts && bash ./patch_ops.sh
\ No newline at end of file