Add missing decorator
Customize decorator to check staff