Update map_decorator.py