Minor improvements on Unlearning.