Adjust style hook for rework
4.0.1