Fix decorartor usage with single argument. Closes #8