only remove from stakinghotkeys if global==0

opentensor · Oct 29, 2024 · fc7d4ef · fc7d4ef
1 parent bac35f9
commit fc7d4ef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pallets/subtensor/src/staking/stake_ops.rs b/pallets/subtensor/src/staking/stake_ops.rs
@@ -395,8 +395,8 @@ impl<T: Config> Pallet<T> {
 
         // Step 15: Decrease total stake across all subnets
         TotalStake::<T>::put(TotalStake::<T>::get().saturating_sub(tao_unstaked_u64));
-        // Step 16: Update StakingHotkeys if the hotkey's total alpha is zero
-        if Alpha::<T>::get((hotkey, coldkey, netuid)) == 0 {
+        // Step 16: Update StakingHotkeys if the hotkey's total alpha, across all subnets, is zero
+        if Self::get_global_for_hotkey_and_coldkey(hotkey, coldkey) == 0 {
             StakingHotkeys::<T>::mutate(coldkey, |hotkeys| {
                 hotkeys.retain(|k| k != hotkey);
             });