Fix session missing bug