diff --git a/integrations/config/config.go b/integrations/config/config.go
index 3674437..36fbef7 100644
--- a/integrations/config/config.go
+++ b/integrations/config/config.go
@@ -1,7 +1,7 @@
 package config
 
 import (
-	"github.com/netbirdio/netbird/management/proto"
+	"github.com/netbirdio/netbird/shared/management/proto"
 	"github.com/netbirdio/netbird/management/server/types"
 )
 
diff --git a/integrations/validator.go b/integrations/validator.go
index f7693e0..cf3375d 100644
--- a/integrations/validator.go
+++ b/integrations/validator.go
@@ -3,7 +3,7 @@ package integrations
 import (
 	"context"
 
-	"github.com/netbirdio/netbird/management/proto"
+	"github.com/netbirdio/netbird/shared/management/proto"
 	"github.com/netbirdio/netbird/management/server/activity"
 	nbpeer "github.com/netbirdio/netbird/management/server/peer"
 	"github.com/netbirdio/netbird/management/server/types"
@@ -44,7 +44,7 @@ func (v *IntegratedValidatorImpl) PeerDeleted(ctx context.Context, _, _ string,
 	return nil
 }
 
-func (v *IntegratedValidatorImpl) SetPeerInvalidationListener(_ func(accountID string)) {
+func (v *IntegratedValidatorImpl) SetPeerInvalidationListener(_ func(accountID string, peerIDs []string)) {
 
 }