I have tried with a class decorator and it failed, that's why I used that.