Fix flexbox styling inconsistency