From 3c7653a37393ad59a676af1bc22a25190210289d Mon Sep 17 00:00:00 2001
From: Michael Birch <michael.birch@aurora.dev>
Date: Wed, 20 Oct 2021 19:27:38 +0200
Subject: [PATCH] Fix bully build

---
 engine/src/lib.rs | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/engine/src/lib.rs b/engine/src/lib.rs
index 392c7e9c2..0020e2bd0 100644
--- a/engine/src/lib.rs
+++ b/engine/src/lib.rs
@@ -489,28 +489,31 @@ mod contract {
     #[cfg(feature = "evm_bully")]
     #[no_mangle]
     pub extern "C" fn begin_chain() {
-        let mut state = Engine::get_state().sdk_unwrap();
+        let mut io = Runtime;
+        let mut state = Engine::get_state(&io).sdk_unwrap();
         require_owner_only(&state);
-        let args: BeginChainArgs = sdk::read_input_borsh().sdk_unwrap();
+        let args: BeginChainArgs = io.read_input_borsh().sdk_unwrap();
         state.chain_id = args.chain_id;
-        Engine::set_state(state);
+        Engine::set_state(&mut io, state);
         // set genesis block balances
         for account_balance in args.genesis_alloc {
             Engine::set_balance(
+                &mut io,
                 &Address(account_balance.address),
                 &crate::prelude::types::Wei::new(U256::from(account_balance.balance)),
             )
         }
         // return new chain ID
-        sdk::return_output(&Engine::get_state().sdk_unwrap().chain_id)
+        io.return_output(&Engine::get_state(&io).sdk_unwrap().chain_id)
     }
 
     #[cfg(feature = "evm_bully")]
     #[no_mangle]
     pub extern "C" fn begin_block() {
-        let state = Engine::get_state().sdk_unwrap();
+        let io = Runtime;
+        let state = Engine::get_state(&io).sdk_unwrap();
         require_owner_only(&state);
-        let _args: BeginBlockArgs = sdk::read_input_borsh().sdk_unwrap();
+        let _args: BeginBlockArgs = io.read_input_borsh().sdk_unwrap();
         // TODO: https://github.com/aurora-is-near/aurora-engine/issues/2
     }