CTxMemPool::removeForBlock now uses RemoveStaged

1 files changed, 6 insertions, 2 deletions

diff --git a/src/txmempool.cpp b/src/txmempool.cpp
index eee6cbf855..01f6c97aea 100644
--- a/src/txmempool.cpp
+++ b/src/txmempool.cpp
@@ -564,8 +564,12 @@ void CTxMemPool::removeForBlock(const std::vector<CTransaction>& vtx, unsigned i
     }
     BOOST_FOREACH(const CTransaction& tx, vtx)
     {
-        std::list<CTransaction> dummy;
-        remove(tx, dummy, false);
+        txiter it = mapTx.find(tx.GetHash());
+        if (it != mapTx.end()) {
+            setEntries stage;
+            stage.insert(it);
+            RemoveStaged(stage);
+        }
         removeConflicts(tx, conflicts);
         ClearPrioritisation(tx.GetHash());
     }