Quick style bug fix