adding update for restore