fix up flexbox