Visual Studio Code color not working when using python types

Question

I am using the new python syntax to describe what types my methods return e.g.,: def method(unpacked_message: dict) -> dict:

This seems to break the