diff --git a/app/api/v1/endpoints/simulation.py b/app/api/v1/endpoints/simulation.py
index 6c4cd46..dac0b84 100644
--- a/app/api/v1/endpoints/simulation.py
+++ b/app/api/v1/endpoints/simulation.py
@@ -280,6 +280,7 @@ async def fastapi_flushing_analysis(
     drainage_node_ID: str = Query(...),
     flush_flow: float = 0,
     duration: int | None = None,
+    scheme_name: str | None = None,
 ) -> str:
     valve_opening = {
         valve_id: float(valves_k[idx]) for idx, valve_id in enumerate(valves)
@@ -291,6 +292,7 @@ async def fastapi_flushing_analysis(
         modify_valve_opening=valve_opening,
         drainage_node_ID=drainage_node_ID,
         flushing_flow=flush_flow,
+        scheme_name=scheme_name,
     )
     return result or "success"