fix set state in withdrawn method

undefinedorgcr · Nov 9, 2024 · 5f7f4af · 5f7f4af
1 parent 47faf2c
commit 5f7f4af
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/contracts/src/fund.cairo b/contracts/src/fund.cairo
@@ -200,7 +200,10 @@ pub mod Fund {
             let caller = get_caller_address();
             let valid_address_1 = contract_address_const::<FundManagerConstants::VALID_ADDRESS_1>();
             let valid_address_2 = contract_address_const::<FundManagerConstants::VALID_ADDRESS_2>();
-            assert!(valid_address_1 == caller || valid_address_2 == caller, "Only Admins can change the fund state.");
+            assert!(
+                valid_address_1 == caller || valid_address_2 == caller,
+                "Only Admins can change the fund state."
+            );
             self.state.write(state);
         }
         fn getState(self: @ContractState) -> u8 {
@@ -224,7 +227,8 @@ pub mod Fund {
             self.token_dispatcher().approve(valid_address, fund_manager_amount);
             self.token_dispatcher().transfer(valid_address, fund_manager_amount);
             assert(self.get_current_goal_state() == 0, 'Pending stks to withdraw');
-            self.setState(4);
+            self.state.write(FundStates::WITHDRAW);
+
             self
                 .emit(
                     DonationWithdraw {