Fixed UI update bug.