Ensure stake fee is correctly applied netuid unit

Author: thewhaleking

bittensor_cli/src/bittensor/subtensor_interface.py

@@ -1534,7 +1534,7 @@ async def get_stake_fee(
         if destination_hotkey_ss58 is not None and destination_netuid is not None:
             destination = (destination_hotkey_ss58, destination_netuid)
 
-        result = await self.query_runtime_api(
+        result_ = await self.query_runtime_api(
             runtime_api="StakeInfoRuntimeApi",
             method="get_stake_fee",
             params=[
@@ -1546,8 +1546,10 @@ async def get_stake_fee(
             ],
             block_hash=block_hash,
         )
+        result = Balance.from_rao(result_)
+        result.set_unit(origin_netuid)
 
-        return Balance.from_rao(result)
+        return result
 
     async def get_scheduled_coldkey_swap(
         self,