Fixed balance bug