Avoid TypeError

TypeError: decorator() takes exactly 2 arguments (3 given)
Merge pull request #255 from myroslav/patch-1

Avoid TypeError