From 663621f79d3f2ccd13019df8057e44db3d1e3fdd Mon Sep 17 00:00:00 2001
From: agatha <agatha@juggalol.com>
Date: Mon, 3 Jun 2024 00:26:20 -0400
Subject: [PATCH] update contract_fund amount

---
 staking/stakecalc2.py | 2 ++
 1 file changed, 2 insertions(+)

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