Fixing highlighting of types in function declaration