How to get function variable to inherit "function declaration" color scheme?

Please see the below screenshot, I want to have the "fn" to be in Yellow color. But it looks like the "fn" is inheriting the "variable" color scheme.

It's a bit annoying that there's no any contrast between a callable stuff and a normally variable.

While in vscode, it just works

I don't believe there's no way to set it in WebStorm, I guess it's just I don't know how to. So could anyone help me make it work? Thanks