Don't lose nformation about a function when using a decorator