Adding fix for removed end to media query