Minor bug fix: support small screen