diff --git a/scripts/simulate.py b/scripts/simulate.py
index 1a4aad2..a56e970 100644
--- a/scripts/simulate.py
+++ b/scripts/simulate.py
@@ -135,9 +135,7 @@ def main(dict_cfg: DictConfig) -> None:
     trained_action_dim = raw_env.action_space.shape[0] // needed_copies
 
     # 7. Load policy
-    message_passing_steps = (
-        metadata.get("architecture", {}) or {}
-    ).get("message_passing_steps")
+    message_passing_steps = (metadata.get("architecture", {}) or {}).get("message_passing_steps")
     if message_passing_steps is None:
         message_passing_steps = 4
     message_passing_steps = int(message_passing_steps)
diff --git a/scripts/train.py b/scripts/train.py
index a6e4fed..c367000 100644
--- a/scripts/train.py
+++ b/scripts/train.py
@@ -8,7 +8,6 @@ from brittle_star_project.configs.register_configs import register_configs
 from brittle_star_project.trainers.PPOTrainer import PPOTrainer
 from brittle_star_project.environment.BrittleStarJaxEnvWrapper import BrittleStarJaxEnvWrapper
 from experiment_logger import init_logger, get_logger
-import logging
 
 
 def make_env(cfg: BrittleStarConfig) -> BrittleStarJaxEnvWrapper: