(Fix) Fixed up bug where balance does not update