fixed restore method