Fix lack of return value in decorator. More PEP8.