Fix bug when session restore