Bug fix in styling