Put back needlessly removed code.