Fix media query bug