diff --git a/cmd/virtual-kubelet/set-version.sh b/cmd/virtual-kubelet/set-version.sh
index bb152a80..4f1d5158 100755
--- a/cmd/virtual-kubelet/set-version.sh
+++ b/cmd/virtual-kubelet/set-version.sh
@@ -2,7 +2,7 @@
 
 if [ -n ${KUBELET_VERSION} ]; then
 cat << EOF > pkg/virtualkubelet/version.go
-package main
+package virtualkubelet
 
 var (
 	kubeletVersion = "$KUBELET_VERSION"