Yeah I agree that this should be set in `inherited` hook. We use this pattern in a lot of places already. It's cleaner and more explicit.