You guys get this to work with vscode terminal?

[tomvonheill]

Looks like my vscode setup is using the zsh terminal, but for some reason not registering vim mode. For the normal terminal this is all working fine, wondering if i'm missing some steps

[tomvonheill]

I'm a fool, forgot to add the import in my zsh shell, just executed it