diff --git a/src/simulations/router.py b/src/simulations/router.py
index 644c5c3..05eac0c 100644
--- a/src/simulations/router.py
+++ b/src/simulations/router.py
@@ -7,6 +7,7 @@ from src.database.core import DbSession
 from src.database.service import CommonParameters
 from src.models import StandardResponse
 from src.simulations.schema import (
+    SimulationBase,
     SimulationCreate,
     SimulationPagination,
     SimulationRead,
@@ -47,7 +48,7 @@ async def get_simulation(db_session: DbSession, simulation_id: str):
     return StandardResponse(data=simulation, message="Data retrieved successfully")
 
 
-@router.post("", response_model=StandardResponse[SimulationRead])
+@router.post("", response_model=StandardResponse[SimulationBase])
 async def create_simulation(
     db_session: DbSession, simulation_in: SimulationCreate, current_user: CurrentUser
 ):
@@ -56,7 +57,7 @@ async def create_simulation(
     return StandardResponse(data=simulation, message="Data created successfully")
 
 
-@router.post("/run", response_model=StandardResponse[SimulationRead])
+@router.post("/run", response_model=StandardResponse[SimulationBase])
 async def run_simulation_endpoint(
     db_session: DbSession,
     payload: SimulationRunPayload,
@@ -80,7 +81,7 @@ async def run_simulation_endpoint(
     )
 
 
-@router.put("/{simulation_id}", response_model=StandardResponse[SimulationRead])
+@router.put("/{simulation_id}", response_model=StandardResponse[SimulationBase])
 async def update_simulation(
     db_session: DbSession,
     simulation_id: str,
@@ -103,7 +104,7 @@ async def update_simulation(
     return StandardResponse(data=updated_simulation, message="Data updated successfully")
 
 
-@router.delete("/{simulation_id}", response_model=StandardResponse[SimulationRead])
+@router.delete("/{simulation_id}", response_model=StandardResponse[SimulationBase])
 async def delete_simulation(db_session: DbSession, simulation_id: str):
     simulation = await get(db_session=db_session, simulation_id=simulation_id)