From 713a50238b82e63fb9a547b6c640d0ec387d444c Mon Sep 17 00:00:00 2001
From: Michiel Vrins <michiel_678@hotmail.com>
Date: Wed, 26 Apr 2023 16:12:20 +0200
Subject: [PATCH] Also attempt to cancel customRegistration during during our
 cancel

---
 example/lib/screens/login_screen.dart | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/example/lib/screens/login_screen.dart b/example/lib/screens/login_screen.dart
index 2f9fd2ad..242df9e8 100644
--- a/example/lib/screens/login_screen.dart
+++ b/example/lib/screens/login_screen.dart
@@ -3,6 +3,7 @@ import 'dart:convert';
 
 import 'package:flutter/material.dart';
 import 'package:flutter/services.dart';
+import 'package:onegini/callbacks/onegini_custom_registration_callback.dart';
 import 'package:onegini/callbacks/onegini_registration_callback.dart';
 import 'package:onegini/events/onewelcome_events.dart';
 import 'package:onegini/model/request_details.dart';
@@ -110,14 +111,14 @@ class _LoginScreenState extends State<LoginScreen> {
 
   cancelRegistration() async {
     setState(() => isLoading = false);
-
-    await OneginiRegistrationCallback()
-        .cancelBrowserRegistration()
-        .catchError((error) {
-      if (error is PlatformException) {
-        showFlutterToast(error.message);
-      }
-    });
+    try {
+      await Future.any([
+        OneginiRegistrationCallback().cancelBrowserRegistration(),
+        OneginiCustomRegistrationCallback().submitErrorAction('Canceled')
+      ]);
+    } on PlatformException catch (error) {
+      showFlutterToast(error.message);
+    }
   }
 
   Future<List<OWUserProfile>> getUserProfiles() async {