Made tiny change to session implementation