diff --git a/pkg/middleware/account_uuid_test.go b/pkg/middleware/account_uuid_test.go
index 7241fa9..3b3a60a 100644
--- a/pkg/middleware/account_uuid_test.go
+++ b/pkg/middleware/account_uuid_test.go
@@ -73,15 +73,15 @@ func TestAccountUUIDMiddlewareWithDisabledAccount(t *testing.T) {
 
 func mockAccountUUIDMiddlewareAccSvc(retErr, accEnabled bool) proto.AccountsService {
 	if retErr {
-		return &proto.MockAccountsService{
-			ListFunc: func(ctx context.Context, in *proto.ListAccountsRequest, opts ...client.CallOption) (out *proto.ListAccountsResponse, err error) {
+		return &proto.AccountsServiceMock{
+			ListAccountsFunc: func(ctx context.Context, in *proto.ListAccountsRequest, opts ...client.CallOption) (out *proto.ListAccountsResponse, err error) {
 				return nil, fmt.Errorf("error returned by mockAccountsService LIST")
 			},
 		}
 	}
 
-	return &proto.MockAccountsService{
-		ListFunc: func(ctx context.Context, in *proto.ListAccountsRequest, opts ...client.CallOption) (out *proto.ListAccountsResponse, err error) {
+	return &proto.AccountsServiceMock{
+		ListAccountsFunc: func(ctx context.Context, in *proto.ListAccountsRequest, opts ...client.CallOption) (out *proto.ListAccountsResponse, err error) {
 			return &proto.ListAccountsResponse{
 				Accounts: []*proto.Account{
 					{