fix layout bug on smaller screen