Make sure we pass on returned values from decorated functions