Update afterSave() signature