src:state.rs: save tree after we generate witness values

src/state.rs

@@ -145,9 +145,6 @@ impl State {
             let node = MerkleNode(coin.0);
             self.tree.append(&node);
 
-            // Save updated merkle tree into wallet.
-            wallet.put_tree(&self.tree).await?;
-
             // Keep track of all Merkle roots that have existed
             self.merkle_roots.put(self.tree.root(), vec![] as Vec<u8>)?;
 
@@ -169,6 +166,8 @@ impl State {
                     }
                 }
             }
+            // Save updated merkle tree into wallet.
+            wallet.put_tree(&self.tree).await?;
         }
 
         trace!("apply() exiting successfully");