You need to use `@wraps` here