Fix some missing highlight groups