diff --git a/management/server/mock_server/account_mock.go b/management/server/mock_server/account_mock.go
index ec29222a46..74557e2275 100644
--- a/management/server/mock_server/account_mock.go
+++ b/management/server/mock_server/account_mock.go
@@ -197,7 +197,7 @@ func (am *MockAccountManager) CreateSetupKey(
 
 // AccountExists mock implementation of AccountExists from server.AccountManager interface
 func (am *MockAccountManager) AccountExists(ctx context.Context, accountID string) (bool, error) {
-	if am.GetAccountIDByUserIdFunc != nil {
+	if am.AccountExistsFunc != nil {
 		return am.AccountExistsFunc(ctx, accountID)
 	}
 	return false, status.Errorf(codes.Unimplemented, "method AccountExists is not implemented")