diff --git a/deploy/models.go b/deploy/models.go
index e19c489..8cc6625 100644
--- a/deploy/models.go
+++ b/deploy/models.go
@@ -50,7 +50,7 @@ func (d *DeployScriptGenerator) GetWekaInstallScript() string {
 	`
 
 	cgroupsMode := "auto"
-	if d.Params.CgroupsMode != "" && d.Params.Protocol == "" {
+	if d.Params.CgroupsMode != "" && (d.Params.Protocol == "" || d.Params.Protocol == protocol.DATA) {
 		cgroupsMode = d.Params.CgroupsMode
 	}