diff --git a/environments/community/ufc_prediction_env/ufc_image_env.py b/environments/community/ufc_prediction_env/ufc_image_env.py
index 8e3959eb..f824e8ce 100644
--- a/environments/community/ufc_prediction_env/ufc_image_env.py
+++ b/environments/community/ufc_prediction_env/ufc_image_env.py
@@ -10,7 +10,12 @@ from typing import List, Optional, Tuple
 from PIL import Image
 from pydantic import Field
 
-from atroposlib.envs.base import BaseEnv, BaseEnvConfig, APIServerConfig, ScoredDataGroup
+from atroposlib.envs.base import (
+    APIServerConfig,
+    BaseEnv,
+    BaseEnvConfig,
+    ScoredDataGroup,
+)
 from atroposlib.type_definitions import GameHistory, Item
 from atroposlib.utils.tokenize_for_trainer import tokenize_for_trainer
 
diff --git a/environments/community/ufc_prediction_env/ufc_server.py b/environments/community/ufc_prediction_env/ufc_server.py
index adc0d1c4..935cad32 100644
--- a/environments/community/ufc_prediction_env/ufc_server.py
+++ b/environments/community/ufc_prediction_env/ufc_server.py
@@ -7,7 +7,12 @@ from typing import List, Optional, Tuple
 
 from pydantic import Field
 
-from atroposlib.envs.base import BaseEnv, BaseEnvConfig, APIServerConfig, ScoredDataGroup
+from atroposlib.envs.base import (
+    APIServerConfig,
+    BaseEnv,
+    BaseEnvConfig,
+    ScoredDataGroup,
+)
 from atroposlib.type_definitions import GameHistory, Item
 from atroposlib.utils.tokenize_for_trainer import tokenize_for_trainer
 
diff --git a/environments/sft_loader_server.py b/environments/sft_loader_server.py
index a8c00863..46277dd5 100644
--- a/environments/sft_loader_server.py
+++ b/environments/sft_loader_server.py
@@ -5,7 +5,12 @@ from typing import Dict, List, Optional, Tuple
 from datasets import load_dataset
 from pydantic import Field
 
-from atroposlib.envs.base import BaseEnv, BaseEnvConfig, APIServerConfig, ScoredDataGroup
+from atroposlib.envs.base import (
+    APIServerConfig,
+    BaseEnv,
+    BaseEnvConfig,
+    ScoredDataGroup,
+)
 from atroposlib.type_definitions import Item