decorator added to setup.py

Hi Tom, 

I think this should fix the travis fail from your pull request. 

Cheers