Update user_context_test.go