diff --git a/atroposlib/envs/server_handling/openai_server.py b/atroposlib/envs/server_handling/openai_server.py
index 3af68e98..33322262 100644
--- a/atroposlib/envs/server_handling/openai_server.py
+++ b/atroposlib/envs/server_handling/openai_server.py
@@ -167,7 +167,7 @@ def resolve_openai_configs(
             message=f"CLI overrides for OpenAI settings (--{openai_full_prefix}*) are not supported "
             f"when multiple servers are defined (either via YAML list under '{OPENAI_NAMESPACE}' "
             "or a default list with length >= 2).",
-            exit_code=2
+            exit_code=2,
         )
 
     if is_multi_server_yaml:
diff --git a/environments/tool_calling_server.py b/environments/tool_calling_server.py
index 7de921aa..a6d8633b 100644
--- a/environments/tool_calling_server.py
+++ b/environments/tool_calling_server.py
@@ -3,10 +3,10 @@ import random
 import re
 from typing import Dict, List, Optional, Tuple, Union
 
-import wandb
 from datasets import load_dataset
 from tqdm.asyncio import tqdm_asyncio
 
+import wandb
 from atroposlib.envs.base import (
     APIServerConfig,
     BaseEnv,