Update session_test.go