Avoid TypeError

TypeError: decorator() takes exactly 2 arguments (3 given)