Small fix for styling