Semantic Tokens are misplaced with language-plugin-pug #4070

Pug performance with the latest language-tools and language-plugin-pug is great, but the new semantic tokens handling seems to be misapplied.

A few visual examples:

In pug

vs HTML

To try and create a minimal reproduction, I've generated the vue-ts vite template and converted the two components to pug templates.
https://stackblitz.com/edit/vitejs-vite-f7dgtx

In HTML we have the HelloWorld component highlighted as expected.

While in pug the semantic token is instead applied over the div and a tags.

Win10
Vscode 1.87.1
@vue/language-plugin-pug 2.0.6