Semantic highlighting of TypeAliases of Unions is not consistent with non unions.

[rchiodo]

For this code here, I get this: (with what's in main and is about to ship):

Number: TypeAlias = int | float

def return_number() -> Number:
    return 1    

But for this code here, I get this:

Number: TypeAlias = int

def return_number() -> Number:
    return 1    

[kytpbs]

Have there been any updates on this? It's making me go crazy