Did you mean to use `@functools.wraps` here?