Fix style bug