LkMEX Transfer: Add energy fix

multiversx · Jan 12, 2024 · 6a45832 · 6a45832
1 parent e4b95af
commit 6a45832
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/locked-asset/lkmex-transfer/src/energy_transfer.rs b/locked-asset/lkmex-transfer/src/energy_transfer.rs
@@ -53,6 +53,7 @@ pub trait EnergyTransferModule:
                 let epoch_diff = current_epoch - attributes.unlock_epoch;
                 let simulated_deplete_amount = &token.amount * epoch_diff;
                 energy.remove_energy_raw(BigUint::zero(), simulated_deplete_amount);
+                energy.add_energy_raw(token.amount, BigInt::zero());
             }
         }