Fixing the collapse on smaller screens