fix: fixes partial withdrawals

tonwhales · Apr 14, 2022 · 3a9959c · 3a9959c
1 parent 14e0510
commit 3a9959c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/compiled/nominators.fc b/compiled/nominators.fc
@@ -852,7 +852,7 @@ int owned_balance() {
 
     ;; Check stake
     throw_unless(error::invalid_stake_value(), stake >= 0);
-    throw_unless(error::invalid_stake_value(), ctx_member_balance + ctx_member_pending_deposit >= stake);
+    throw_unless(error::invalid_stake_value(), ctx_member_balance + ctx_member_withdraw + ctx_member_pending_deposit >= stake);
 
     ;; Try to withdraw immediatelly
     var (withdrawed, all) = member_stake_withdraw(stake);

diff --git a/sources/modules/op-common.fc b/sources/modules/op-common.fc
@@ -61,7 +61,7 @@
 
     ;; Check stake
     throw_unless(error::invalid_stake_value(), stake >= 0);
-    throw_unless(error::invalid_stake_value(), ctx_member_balance + ctx_member_pending_deposit >= stake);
+    throw_unless(error::invalid_stake_value(), ctx_member_balance + ctx_member_withdraw + ctx_member_pending_deposit >= stake);
 
     ;; Try to withdraw immediatelly
     var (withdrawed, all) = member_stake_withdraw(stake);