Visual Studio Code: highlighted imports in python

I use Visual Studio Code both on my work Windows laptop, and on my Ubuntu PC at home.

There's a functionality that I can see from my Windows vscode that I'd like to have on my Ubuntu one as well, but can't find whether it's a setting, extension, or something else.

The functionality is this: python imports are highlighted in green, and if they are not used, they are dimmed.

Windows screenshot, how I want it to work:

Ubuntu screenshot, how it currently is:

Does anyone know where I should look? I'm using the same colour theme on both, Monokai.

Thanks!