diff --git a/Dockerfile b/Dockerfile
index f2e6bdb01..f98b8deff 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -11,10 +11,10 @@ RUN go mod download
 # Do actual build
 COPY . /go/node-feature-discovery
 
-ARG NFD_VERSION
+ARG VERSION
 ARG HOSTMOUNT_PREFIX
 
-RUN make install VERSION=$NFD_VERSION HOSTMOUNT_PREFIX=$HOSTMOUNT_PREFIX
+RUN make install VERSION=$VERSION HOSTMOUNT_PREFIX=$HOSTMOUNT_PREFIX
 
 RUN make test
 
diff --git a/Makefile b/Makefile
index 4f8738847..1de4199d7 100644
--- a/Makefile
+++ b/Makefile
@@ -44,7 +44,7 @@ install:
 	$(GO_CMD) install -v $(LDFLAGS) ./cmd/...
 
 image: yamls
-	$(IMAGE_BUILD_CMD) --build-arg NFD_VERSION=$(VERSION) \
+	$(IMAGE_BUILD_CMD) --build-arg VERSION=$(VERSION) \
 		--build-arg HOSTMOUNT_PREFIX=$(CONTAINER_HOSTMOUNT_PREFIX) \
 		-t $(IMAGE_TAG) \
 		$(IMAGE_BUILD_EXTRA_OPTS) ./