Fixing flexbox issue