From 0bf13096d74e84a6a3dbf41996f18242877d6632 Mon Sep 17 00:00:00 2001
From: Ben Mason <ben@wattmason.co.uk>
Date: Sun, 19 Jan 2025 14:31:55 +0100
Subject: [PATCH] Update pallets/subtensor/src/staking/transfer_stake.rs

Co-authored-by: Cameron Fairchild <cameron@fairchild.dev>
---
 pallets/subtensor/src/staking/transfer_stake.rs | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/pallets/subtensor/src/staking/transfer_stake.rs b/pallets/subtensor/src/staking/transfer_stake.rs
index 3bad491cc..29bfde72a 100644
--- a/pallets/subtensor/src/staking/transfer_stake.rs
+++ b/pallets/subtensor/src/staking/transfer_stake.rs
@@ -41,10 +41,8 @@ impl<T: Config> Pallet<T> {
         );
 
         // --- 4. Get the current alpha stake for the origin hotkey-coldkey pair in the origin subnet
-        let origin_alpha =
-            Self::get_stake_for_hotkey_and_coldkey_on_subnet(&hotkey, &coldkey, netuid);
         ensure!(
-            alpha_amount <= origin_alpha,
+   Self::has_enough_stake_on_subnet(&hotkey, &coldkey, netuid, alpha_amount)
             Error::<T>::NotEnoughStakeToWithdraw
         );