diff --git a/.dockerignore b/.dockerignore
index f72ab56..0fd63c1 100644
--- a/.dockerignore
+++ b/.dockerignore
@@ -46,5 +46,4 @@ Dockerfile
 *.log
 
 # Testing
-test_*.py
 *_test.py
diff --git a/main.py b/main.py
index a2ab11d..800ddcf 100644
--- a/main.py
+++ b/main.py
@@ -1480,7 +1480,7 @@ async def health_check():
     """Health check endpoint"""
     return HealthCheckResponse(status="healthy", message="Soil water balance model API is running")
 
-@app.post("/simulate", response_model=Dict[str, any])
+@app.post("/simulate", response_model=Dict[str, Any])
 async def simulate_soil_water_balance(request: SimulationRequest):
     """
     Run soil water balance simulation