Add context to decorators