Fixed balance updating on loading tests