Add tests for decorators