Restore changes after testing