diff --git a/tests/test-server/override_logging.py b/tests/test-server/override_logging.py
index 41be4e32..b29cf1b4 100644
--- a/tests/test-server/override_logging.py
+++ b/tests/test-server/override_logging.py
@@ -151,3 +151,5 @@ def transform_logged_data(data: Any, visited: Union[Set[Any], None] = None) -> A
         return data.to_json()
     if isinstance(data, IsVerifiedSCV):
         return "IsVerifiedSCV"
+
+    return data
diff --git a/tests/test-server/test_functions_mapper.py b/tests/test-server/test_functions_mapper.py
index 5f76f915..df95a83a 100644
--- a/tests/test-server/test_functions_mapper.py
+++ b/tests/test-server/test_functions_mapper.py
@@ -346,8 +346,8 @@ async def consume_code_post(
             ) -> Any:
                 o = await api_options.request.json()
                 assert o is not None
-                if o.get("user_context", {}).get("DO_LINK") is not None:
-                    user_context["DO_LINK"] = o["user_context"]["DO_LINK"]
+                if o.get("userContext", {}).get("DO_LINK") is not None:
+                    user_context["DO_LINK"] = o["userContext"]["DO_LINK"]
                 return await og(
                     pre_auth_session_id,
                     user_input_code,
@@ -417,8 +417,8 @@ async def sign_up_post(
                 if "signUpPOST" in eval_str:
                     n = await api_options.request.json()
                     assert n is not None
-                    if n.get("user_context", {}).get("DO_LINK") is not None:
-                        user_context["DO_LINK"] = n["user_context"]["DO_LINK"]
+                    if n.get("userContext", {}).get("DO_LINK") is not None:
+                        user_context["DO_LINK"] = n["userContext"]["DO_LINK"]
                 return await og_sign_up_post(
                     form_fields,
                     tenant_id,