Guess the decorator doesn't work the way I wanted