modify decorator.py
modify decorator.py