Fix integration test for session