fixed restore func.