diff --git a/atroposlib/envs/server_handling/openai_server.py b/atroposlib/envs/server_handling/openai_server.py
index e54a295b..cc541d7c 100644
--- a/atroposlib/envs/server_handling/openai_server.py
+++ b/atroposlib/envs/server_handling/openai_server.py
@@ -37,8 +37,8 @@ class OpenAIServer(APIServer):
                     await self.openai.completions.create(
                         model=self.config.model_name,
                         prompt="hi",
-                    max_tokens=1,
-                )
+                        max_tokens=1,
+                    )
                 self.server_healthy = True
             except (
                 aiohttp.ClientError,
diff --git a/atroposlib/envs/server_handling/server_baseline.py b/atroposlib/envs/server_handling/server_baseline.py
index 179a6a5c..daa82415 100644
--- a/atroposlib/envs/server_handling/server_baseline.py
+++ b/atroposlib/envs/server_handling/server_baseline.py
@@ -320,7 +320,9 @@ class APIServer(ABC):
             if (
                 self.config.base_url is not None
             ):  # skip health check if using OpenAI API
-                self.check_task = asyncio.create_task(self.check_server_status_task(chat_completion=False))
+                self.check_task = asyncio.create_task(
+                    self.check_server_status_task(chat_completion=False)
+                )
             else:
                 self.server_healthy = True
             self.initialized = True