Theme does not apply immediately when loading via extension

[dhrubomoy]

Syntax highlighting does not apply after the editor is rendered. But if I interact with the editor e.g. type something or scroll, only then it gets applied. I am applying custom theme that I am loading via extension in vscodeApiConfig (code here). I am loading the worker that I bundled using webpack like its mentioned in this doc.

The issue does not happen when I set 'editor.experimental.asyncTokenization': false

You can reproduce the issue in my branch here

npm install
npm run build
npm run bundle:workers --workspace packages/examples
npm run dev

Then open the statemachine react example in browser