Fixing import wildcard highlighting