The color is added depending on the language. You can check whether Swift is added in the highlighted_languages and if detection still does not work, you can also manually set the language for a code block.