UI fix for small screen devices