Fix dismiss button on small screens