fixed .restore() functionality