Styling bug fix