fix bug on small screen