diff --git a/graphql/e2e/custom_logic/cmd/Dockerfile b/graphql/e2e/custom_logic/cmd/Dockerfile
index be8fc4f5d4a..2ff6778d93b 100644
--- a/graphql/e2e/custom_logic/cmd/Dockerfile
+++ b/graphql/e2e/custom_logic/cmd/Dockerfile
@@ -5,9 +5,12 @@ COPY . .
 
 RUN apk update && apk add git && apk add nodejs && apk add npm
 
-RUN go install gopkg.in/yaml.v2@latest
+# As of Go 1.16 modules are on by default
+RUN go env -w GO111MODULE=auto
 
-RUN go install github.com/graph-gophers/graphql-go@latest
+RUN go get gopkg.in/yaml.v2
+
+RUN go get github.com/graph-gophers/graphql-go
 
 RUN npm install