Fix tree node not change after save changes on prompt clicks (for AO update)