diff --git a/src/toldg/predict.py b/src/toldg/predict.py
index bf55ad4..1865b27 100644
--- a/src/toldg/predict.py
+++ b/src/toldg/predict.py
@@ -239,7 +239,7 @@ class TransactionClassifier:
             output_dim=len(self.categories),
             learning_rate=0.05,
             reg_lambda=0.01,
-            max_iterations=3000,
+            max_iterations=6000,
         )
 
         self.model.train(x_vectors, label_indices, verbose=verbose)
diff --git a/src/toldg/process.py b/src/toldg/process.py
index 6d98115..33bae46 100644
--- a/src/toldg/process.py
+++ b/src/toldg/process.py
@@ -76,6 +76,7 @@ def apply_mappings(transactions: list[Transaction], mappings: dict[str, Mapping]
             assert mapping.count > 0, f"{mapping} used by {t} but count is not greater than '0'."
             mapping.count -= 1
             t.mapping = mapping
+            t.account2 = mapping.account2
         else:
             unmapped_count += 1
     if unmapped_count > 0: