Fix error in afterSave()