To be consistent, I'd like this feature to work when using the decorators in ```hooks.py``` as well.