Wizard add diagnosis correctly restored