diff --git a/src/tagstudio/qt/mixed/migration_modal.py b/src/tagstudio/qt/mixed/migration_modal.py
index 77850a72..def95daf 100644
--- a/src/tagstudio/qt/mixed/migration_modal.py
+++ b/src/tagstudio/qt/mixed/migration_modal.py
@@ -361,20 +361,16 @@ class JsonMigrationModal(QObject):
 
         try:
             iterator = FunctionIterator(self.migration_iterator)
-            iterator.value.connect(
-                lambda x: (
-                    pb.setLabelText(header(x, 4)),
-                    self.update_sql_value_ui(show_msg_box=False)
-                    if x == Translations["json_migration.checking_for_parity"]
-                    else (),
-                    self.update_parity_ui()
-                    if x == Translations["json_migration.checking_for_parity"]
-                    else (),
-                )
-            )
+
+            if skip_ui:
+                iterator.run()
+                return
+
+            iterator.value.connect(lambda x: pb.setLabelText(header(x, 4)))
             r = CustomRunnable(iterator.run)
             r.done.connect(
                 lambda: (
+                    self.update_parity_ui(),
                     self.update_sql_value_ui(show_msg_box=not skip_ui),
                     pb.setMinimum(1),
                     pb.setValue(1),
@@ -423,7 +419,6 @@ class JsonMigrationModal(QObject):
                 yield Translations["json_migration.migration_complete"]
             else:
                 yield Translations["json_migration.migration_complete_with_discrepancies"]
-            self.update_parity_ui()
             QApplication.beep()
             QApplication.alert(self.paged_panel)
             self.done = True
diff --git a/tests/test_json_migration.py b/tests/test_json_migration.py
index 3c3c15db..003eb08f 100644
--- a/tests/test_json_migration.py
+++ b/tests/test_json_migration.py
@@ -3,7 +3,6 @@
 
 
 from pathlib import Path
-from time import time
 
 from tagstudio.qt.mixed.migration_modal import JsonMigrationModal
 
@@ -14,10 +13,6 @@ def test_json_migration():
     modal = JsonMigrationModal(CWD.parent / "fixtures" / "json_library")
     modal.migrate(skip_ui=True)
 
-    start = time()
-    while not modal.done and (time() - start < 60):
-        pass
-
     # Entries ==================================================================
     # Count
     assert len(modal.json_lib.entries) == modal.sql_lib.entries_count