What is shown as the wallet balance is actually the "transferable balance". The reason we show this is in order to show the user the amount of tokens that can be used at the specific time. A staked token is never moved out of your wallet (just not shown as part of your current balance))