restore() function bug fix