diff --git a/tools/server/server.py b/tools/server/server.py
index 09ded0d9..849db89c 100644
--- a/tools/server/server.py
+++ b/tools/server/server.py
@@ -74,7 +74,7 @@ def create_app(config: ServerConfig) -> FastAPI:
         return {"status": "deleted"}
 
     @app.get("/experiments/{name}/batch", response_model=BatchResponse)
-    async def generate_batch(name: str, base_index: int, batch_size: int):
+    async def generate_batch(name: str, base_index: int, batch_size: int, epoch: int = 0):
         """Generate a batch of raw entries"""
         # Validate parameters
         if base_index < 0: