diff --git a/environments/community/philosophical_rlaif_env.py b/environments/community/philosophical_rlaif_env.py
index b07df44d..04d98266 100644
--- a/environments/community/philosophical_rlaif_env.py
+++ b/environments/community/philosophical_rlaif_env.py
@@ -3,10 +3,10 @@ import random
 import re  # Added import for regular expressions
 from typing import Dict, List, Optional, Tuple
 
+import wandb  # type: ignore
 from datasets import load_dataset  # type: ignore
 from pydantic import Field
 
-import wandb  # type: ignore
 from atroposlib.envs.base import BaseEnvConfig  # Ensure BaseEnvConfig is imported
 from atroposlib.envs.base import (
     APIServerConfig,
diff --git a/environments/tool_calling_server.py b/environments/tool_calling_server.py
index a6d8633b..7de921aa 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,