Small improvement to media query issue (causing black screen)