Restore lost code after update