Fix flexbox layout