fix: decorators.py

Make sure that we include all the args when we return the function