Minor fix in the session