diff --git a/usecases/cem/vabd/testhelper_test.go b/usecases/cem/vabd/testhelper_test.go
index 067fde71..e067e486 100644
--- a/usecases/cem/vabd/testhelper_test.go
+++ b/usecases/cem/vabd/testhelper_test.go
@@ -156,7 +156,7 @@ func setupDevices(
 						Device: util.Ptr(model.AddressDeviceType(remoteDeviceName)),
 						Entity: []model.AddressEntityType{1},
 					},
-					EntityType: util.Ptr(model.EntityTypeTypeElectricityStorageSystem),
+					EntityType: util.Ptr(model.EntityTypeTypeBatterySystem),
 				},
 			},
 		},