fix small screen behaviour