fix flexbox issue