diff --git a/environments/code_execution_server/coding_server.py b/environments/code_execution_server/coding_server.py
index 43581a07..0a64acf3 100644
--- a/environments/code_execution_server/coding_server.py
+++ b/environments/code_execution_server/coding_server.py
@@ -666,9 +666,7 @@ class CodingEnv(BaseEnv):
     async def setup(self):
         """Setup the environment"""
         if self.config.dataset_name == "deepmind":
-            self.train = load_dataset(
-                "deepmind/code_contests", split="train"
-            )  # CHANGE
+            self.train = load_dataset("deepmind/code_contests", split="train")  # CHANGE
         else:
             self.train = load_dataset(
                 "NousResearch/RLVR_Coding_Problems", split="train"