Fix flexbox issues with code blocks in Safari