diff --git a/tensorflow_runtime_dockerfiles/gpu.Dockerfile b/tensorflow_runtime_dockerfiles/gpu.Dockerfile
index 906ab3389b90390cc7eaadcaa369b90836ab1b17..73512d401c48a094ba9d9265fd53f3f68877ead9 100644
--- a/tensorflow_runtime_dockerfiles/gpu.Dockerfile
+++ b/tensorflow_runtime_dockerfiles/gpu.Dockerfile
@@ -52,5 +52,5 @@ CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/
 FROM base as test
 
 ENV LD_LIBRARY_PATH /usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH
-COPY test.import_gpu.sh /test.import_gpu.sh
-RUN /test.import_gpu.sh
+COPY test.import_cpu.sh /test.import_cpu.sh
+RUN /test.import_cpu.sh