Make restore atomic