Session small fix