Fix CSS for flex display changes