Highlighting works in unsaved file, disappears in saved file

[henrebotha]

If I create a file and use TODO somewhere in the middle of a sentence, it highlights correctly. If I then save the file and reopen it, the highlighting disappears. See GIF

[ipleten]

I noticed that this doesn't work for at least puppet syntax. And works as expected for yaml.