diff --git a/src/api/settings.py b/src/api/settings.py
index 19cbf1e3..b4a15745 100644
--- a/src/api/settings.py
+++ b/src/api/settings.py
@@ -633,8 +633,8 @@ async def onboarding(request, flows_service):
 
                 await flows_service.change_langflow_model_value(
                     provider,
-                    body["embedding_model"],
-                    body["llm_model"],
+                    body.get("embedding_model"),
+                    body.get("llm_model"),
                     body.get("endpoint"),
                 )
 
diff --git a/tests/conftest.py b/tests/conftest.py
index eadca98b..e052d03c 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -50,7 +50,6 @@ async def onboard_system():
             "model_provider": "openai",
             "embedding_model": "text-embedding-3-small",
             "llm_model": "gpt-4o-mini",
-            "endpoint": "https://api.openai.com/v1",
             "sample_data": False,
         }
         resp = await client.post("/onboarding", json=onboarding_payload)