diff --git a/images/virtualization-artifact/werf.inc.yaml b/images/virtualization-artifact/werf.inc.yaml
index ad4e475ff..1f23c8f16 100644
--- a/images/virtualization-artifact/werf.inc.yaml
+++ b/images/virtualization-artifact/werf.inc.yaml
@@ -32,5 +32,5 @@ shell:
   - export GOOS=linux
   - export CGO_ENABLED=0
   - export GOARCH=amd64
-  - go build -tags EE -v -a -o virtualization-controller ./cmd/virtualization-controller
+  - go build -tags {{ .MODULE_EDITION }} -v -a -o virtualization-controller ./cmd/virtualization-controller
   - go build -v -a -o virtualization-api ./cmd/virtualization-api