Restore session fix