Staking portal change recommendation

So when you undelegate it takes that ammount out of your total one count until the one is either re-delegated or makes it fully back into your wallet. This was brought to my attention by a delegator that was spooked that he possibly waw missing coins he undelegated.

In the attached picture this wallet actually has close to 630k one, but with the 50k being undelegated it shows 579k.

Hoping this crowd could either fix this or answer if it is feasible to fix it.
Screenshots_2021-09-11-17-13-40