diff --git a/Vagrantfile b/Vagrantfile
index 317fd28e8..f3430bd01 100644
--- a/Vagrantfile
+++ b/Vagrantfile
@@ -154,7 +154,8 @@ EOF
         source /etc/environment
         source /etc/profile.d/sh.local
         set -eux -o pipefail
-        ${GOPATH}/src/github.com/containerd/containerd/script/setup/install-cni
+        cd ${GOPATH}/src/github.com/containerd/containerd
+        script/setup/install-cni
         PATH=/opt/cni/bin:$PATH type ${CNI_BINARIES} || true
     SHELL
   end
diff --git a/contrib/Dockerfile.test b/contrib/Dockerfile.test
index 22a35a5e5..dd8a11003 100644
--- a/contrib/Dockerfile.test
+++ b/contrib/Dockerfile.test
@@ -40,9 +40,8 @@ WORKDIR /go/src/github.com/containerd/containerd
 
 FROM golang AS cni
 ENV DESTDIR=/build
-COPY script/setup/install-cni ./
-COPY go.mod /go/src/github.com/containerd/containerd/go.mod
-RUN ./install-cni
+COPY script/setup/install-cni go.mod /
+RUN DESTDIR=/build /install-cni
 
 FROM golang AS critools
 ARG DESTDIR=/build
diff --git a/script/setup/install-cni b/script/setup/install-cni
index 3bd7ef89a..63a9ba6ac 100755
--- a/script/setup/install-cni
+++ b/script/setup/install-cni
@@ -21,7 +21,7 @@
 #
 set -eu -o pipefail
 
-CNI_COMMIT=${1:-$(grep containernetworking/plugins "$GOPATH"/src/github.com/containerd/containerd/go.mod | awk '{print $2}')}
+CNI_COMMIT=${1:-$(go list -f "{{.Version}}" -m github.com/containernetworking/plugins)}
 CNI_DIR=${DESTDIR:=''}/opt/cni
 CNI_CONFIG_DIR=${DESTDIR}/etc/cni/net.d