Modified the unit-test for session.