Fixed problem with media queries