Fix media query to follow new layout width