[FIX] Balance pie bug fixes