Style bug fix