Small fix in session class