fix: properly set context env (#1070)

bluealloy · Feb 10, 2024 · 52c517c · 52c517c
1 parent bbca128
commit 52c517c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/crates/revm/src/builder.rs b/crates/revm/src/builder.rs
@@ -101,14 +101,14 @@ impl<'a, EXT, DB: Database> EvmBuilder<'a, SetGenericStage, EXT, DB> {
 
     /// Sets Builder with [`EnvWithHandlerCfg`].
     pub fn with_env_with_handler_cfg(
-        self,
+        mut self,
         env_with_handler_cfg: EnvWithHandlerCfg,
     ) -> EvmBuilder<'a, HandlerStage, EXT, DB> {
+        let EnvWithHandlerCfg { env, handler_cfg } = env_with_handler_cfg;
+        self.context.evm.env = env;
         EvmBuilder {
             context: self.context,
-            handler: EvmBuilder::<'a, HandlerStage, EXT, DB>::handler(
-                env_with_handler_cfg.handler_cfg,
-            ),
+            handler: EvmBuilder::<'a, HandlerStage, EXT, DB>::handler(handler_cfg),
             phantom: PhantomData,
         }
     }