Fix onSave hook.