diff --git a/pkg/mocks/mockOpenOltClient.go b/pkg/mocks/mockOpenOltClient.go
index faca16e..5df754f 100644
--- a/pkg/mocks/mockOpenOltClient.go
+++ b/pkg/mocks/mockOpenOltClient.go
@@ -22,6 +22,7 @@
 	"errors"
 	"io"
 
+	config "github.com/opencord/voltha-protos/v3/go/ext/config"
 	openolt "github.com/opencord/voltha-protos/v3/go/openolt"
 	tech_profile "github.com/opencord/voltha-protos/v3/go/tech_profile"
 	"google.golang.org/grpc"
@@ -244,7 +245,7 @@
 }
 
 // OnuItuPonAlarmSet mocks the OnuItuPonAlarmSet function of Openoltclient.
-func (ooc *MockOpenoltClient) OnuItuPonAlarmSet(ctx context.Context, in *openolt.OnuItuPonAlarm, opts ...grpc.CallOption) (*openolt.Empty, error) {
+func (ooc *MockOpenoltClient) OnuItuPonAlarmSet(ctx context.Context, in *config.OnuItuPonAlarm, opts ...grpc.CallOption) (*openolt.Empty, error) {
 	return &openolt.Empty{}, nil
 }