Fix after_remove callback.