diff --git a/libs/remix-lib/src/execution/txRunnerVM.ts b/libs/remix-lib/src/execution/txRunnerVM.ts
index ffdb00e6391..da5ad0e7b2f 100644
--- a/libs/remix-lib/src/execution/txRunnerVM.ts
+++ b/libs/remix-lib/src/execution/txRunnerVM.ts
@@ -140,9 +140,9 @@ export class TxRunnerVM {
           callback(err, result)
         })
       } else {
-        await this.getVMObject().vm.evm.journal.checkpoint()
-        this.runTxInVm(tx, block, async (err, result) => {
-          await this.getVMObject().vm.evm.journal.revert()
+        const root = await this.getVMObject().stateManager.getStateRoot()
+        this.runBlockInVm(tx, block, async (err, result) => {
+          await this.getVMObject().stateManager.setStateRoot(root)
           callback(err, result)
         })
       }