I think this line should probably be in Restore(...) also