minor fixes

heliaxdev · Apr 12, 2024 · 0ca51e5 · 0ca51e5
1 parent 4f89952
commit 0ca51e5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/gen/state.rs b/src/gen/state.rs
@@ -377,9 +377,9 @@ impl State {
         tx_settings: &Option<TxSettings>,
     ) {
         let gas_limit = if let Some(tx_settings) = tx_settings {
-            tx_settings.gas_limit
+            tx_settings.gas_limit + (1 * NATIVE_SCALE)
         } else {
-            DEFAULT_GAS_LIMIT
+            DEFAULT_GAS_LIMIT + (1 * NATIVE_SCALE)
         };
         let gas_price = (gas_limit as f64 * DEFAULT_GAS_PRICE * NATIVE_SCALE as f64).ceil() as u64;
         self.decrease_account_token_balance(address_alias, &Alias::native_token(), gas_price)

diff --git a/src/gen/steps/withdraw.rs b/src/gen/steps/withdraw.rs
@@ -34,7 +34,7 @@ impl Step for Withdraw {
 
     fn update_state(&self, state: &mut crate::state::State) {
         state.insert_withdraw(&self.source, self.amount, self.unbond_step);
-        state.decrease_account_fees(&self.source, &None);
+        state.decrease_account_fees(&self.tx_settings.gas_payer, &None);
     }
 
     fn post_hooks(&self, step_index: u64, _state: &State) -> Vec<Box<dyn crate::step::Hook>> {