Restore code after overwrite