Fix small screen error