Fix flex for small screens.