Add decorator test