Fix decorator return value. Useful for development.
Bump version to 0.2.1