
This is a great feature, but unfortunately it works only with rasterized layers.
For a non-destructive workflow it should either work with Smart objects or convert its output to a Curves adjustment layer. Since any color correction can be done with the Curves, I'm pretty sure it's possible 🙂