I suggest to add an update balance feature which will monitor changings on the wallet. For example when you collect your funds and want to re-invest them in the same pool firstly you need to click on claim then refresh the page to be able to see balance on site and only then you can stake them back.