From 2b8c70ef9e311acfc640dd0f5c74a14d8c0abad1 Mon Sep 17 00:00:00 2001
From: Matthew Chen <matthew@signal.org>
Date: Tue, 16 Oct 2018 16:44:34 -0400
Subject: [PATCH] Fix UD attributes migration completion.

---
 .../migrations/OWS111UDAttributesMigration.swift       | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/SignalMessaging/environment/migrations/OWS111UDAttributesMigration.swift b/SignalMessaging/environment/migrations/OWS111UDAttributesMigration.swift
index 96dec7ff8..831599f53 100644
--- a/SignalMessaging/environment/migrations/OWS111UDAttributesMigration.swift
+++ b/SignalMessaging/environment/migrations/OWS111UDAttributesMigration.swift
@@ -26,15 +26,13 @@ public class OWS111UDAttributesMigration: OWSDatabaseMigration {
 
     override public func runUp(completion: @escaping OWSDatabaseMigrationCompletion) {
         Logger.debug("")
-        BenchAsync(title: "UD Attributes Migration") { completeBenchmark in
-            self.doMigration {
-                completeBenchmark()
-                completion()
-            }
+        Bench(title: "UD Attributes Migration") {
+            self.doMigration()
         }
+        completion()
     }
 
-    private func doMigration(completion: @escaping OWSDatabaseMigrationCompletion) {
+    private func doMigration() {
         tsAccountManager.updateAccountAttributes()
 
         self.dbReadWriteConnection().readWrite { transaction in