AsyncBlockOperation: override finish()

1 files changed, 8 insertions, 7 deletions

diff --git a/ios/MullvadVPN/Operations/AsyncBlockOperation.swift b/ios/MullvadVPN/Operations/AsyncBlockOperation.swift
index c02577471f..66a6d2aa0b 100644
--- a/ios/MullvadVPN/Operations/AsyncBlockOperation.swift
+++ b/ios/MullvadVPN/Operations/AsyncBlockOperation.swift
@@ -24,6 +24,14 @@ class AsyncBlockOperation: AsyncOperation {
         executionBlock = nil
     }
 
+    override func finish() {
+        stateLock.lock()
+        cancellationBlocks.removeAll()
+        stateLock.unlock()
+
+        super.finish()
+    }
+
     override func cancel() {
         super.cancel()
 
@@ -39,7 +47,6 @@ class AsyncBlockOperation: AsyncOperation {
 
     func addCancellationBlock(_ block: @escaping () -> Void) {
         stateLock.lock()
-
         if isCancelled {
             stateLock.unlock()
             block()
@@ -48,10 +55,4 @@ class AsyncBlockOperation: AsyncOperation {
             stateLock.unlock()
         }
     }
-
-    override func operationDidFinish() {
-        stateLock.lock()
-        cancellationBlocks.removeAll()
-        stateLock.unlock()
-    }
 }