Fix restore script