diff --git a/internal/server/boltdbstate/add_on.go b/internal/server/boltdbstate/add_on.go
index 79db8c8191b..1ff39528dcd 100644
--- a/internal/server/boltdbstate/add_on.go
+++ b/internal/server/boltdbstate/add_on.go
@@ -36,7 +36,7 @@ func (s *State) AddOnPut(ctx context.Context, addOn *pb.AddOn) (*pb.AddOn, error
 	return nil, status.Errorf(codes.Unimplemented, "Add On Unimplemented")
 }
 
-func (s *State) AddOnUpdate(ctx context.Context, addOn *pb.AddOn) (*pb.AddOn, error) {
+func (s *State) AddOnUpdate(ctx context.Context, addOn *pb.AddOn, existingAddOn *pb.Ref_AddOn) (*pb.AddOn, error) {
 	return nil, status.Errorf(codes.Unimplemented, "Add On Unimplemented")
 }
 
diff --git a/pkg/serverstate/serverstate.go b/pkg/serverstate/serverstate.go
index ff135ee1ca1..ee5cc92c2d8 100644
--- a/pkg/serverstate/serverstate.go
+++ b/pkg/serverstate/serverstate.go
@@ -236,6 +236,7 @@ type Interface interface {
 
 	AddOnPut(ctx context.Context, addOn *pb.AddOn) (*pb.AddOn, error)
 	AddOnGet(ctx context.Context, addOn *pb.Ref_AddOn) (*pb.AddOn, error)
+	AddOnUpdate(ctx context.Context, addOn *pb.AddOn, existingAddOn *pb.Ref_AddOn) (*pb.AddOn, error)
 	AddOnDelete(ctx context.Context, addOn *pb.Ref_AddOn) error
 	AddOnList(ctx context.Context, request *pb.ListAddOnsRequest) ([]*pb.AddOn, *pb.PaginationResponse, error)
 }