diff --git a/internal/app/service/device/general_device_service.go b/internal/app/service/device/general_device_service.go
index b3caeed..a5cdade 100644
--- a/internal/app/service/device/general_device_service.go
+++ b/internal/app/service/device/general_device_service.go
@@ -2,9 +2,11 @@ package device
 
 import (
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/transport"
 )
 
 type GeneralDeviceService struct {
+	comm transport.Communicator
 }
 
 func (g *GeneralDeviceService) Switch(device models.Device, action DeviceAction) error {