I guess you're right, moved the code to inside the decorator.