Balance pie bug fixes