Added two more tests for the decorator