diff --git a/staking/stakecalc2.py b/staking/stakecalc2.py
index 6231233..9d296fc 100644
--- a/staking/stakecalc2.py
+++ b/staking/stakecalc2.py
@@ -32,6 +32,8 @@ class StakingPool:
         self.__distribute_reward(reward_amount)
 
         # Deposit amount minus fee to contract_fund (wallet)
+        self.contract_fund += amount - total_fee
+
         # Update user_balances with deposit amount
         if user not in self.user_balances:
             self.user_balances[user] = amount - total_fee