Update test
Add decorator