fix the decorator