Update stream-android-core/src/main/java/io/getstream/android/core/api/utils/Result.kt

Co-authored-by: Copilot <[email protected]>

Author: aleksandar-apostolov

stream-android-core/src/main/java/io/getstream/android/core/api/utils/Result.kt

@@ -49,10 +49,13 @@ public inline fun <T, R> Result<T>.flatMap(transform: (T) -> Result<R>): Result<
 @StreamInternalApi
 public suspend fun <T> Result<T>.onTokenError(
     function: suspend (exception: StreamEndpointException, code: Int) -> Result<T>
-): Result<T> = onFailure {
-    if (it is StreamEndpointException && it.apiError?.code?.div(10) == 4) {
-        function(it, it.apiError.code)
-    } else {
-        this
+): Result<T> = fold(
+    onSuccess = { this },
+    onFailure = { throwable ->
+        if (throwable is StreamEndpointException && throwable.apiError?.code?.div(10) == 4) {
+            function(throwable, throwable.apiError.code)
+        } else {
+            this
+        }
     }
-}
+)