fix flexbox bug