diff --git a/go_crypto/Dockerfile b/go_crypto/Dockerfile
index 03a7cc9..70f537d 100644
--- a/go_crypto/Dockerfile
+++ b/go_crypto/Dockerfile
@@ -3,6 +3,7 @@ From ghcr.io/samyuan1990/liboqs-go:latest
 RUN mkdir -p /app
 WORKDIR /app
 COPY main.go .
+COPY main_test.go .
 COPY go.mod .
 
 #ENV GOPROXY=https://goproxy.cn,direct
diff --git a/go_pqc/Dockerfile b/go_pqc/Dockerfile
index 782f714..3a17d73 100644
--- a/go_pqc/Dockerfile
+++ b/go_pqc/Dockerfile
@@ -3,6 +3,7 @@ From ghcr.io/samyuan1990/liboqs-go:latest
 RUN mkdir -p /app
 WORKDIR /app
 COPY main.go .
+COPY main_test.go .
 COPY go.mod .
 COPY go.sum .
 COPY vendor .