diff --git a/environments/community/lean_proof_env/lean_env.py b/environments/community/lean_proof_env/lean_env.py
index bb6acb55..f9c5660b 100644
--- a/environments/community/lean_proof_env/lean_env.py
+++ b/environments/community/lean_proof_env/lean_env.py
@@ -2,10 +2,10 @@ import asyncio  # For async operations if PyPantograph is async
 import random
 from typing import Dict, List, Optional, Tuple, TypedDict
 
+import wandb  # For wandb.Table
 from datasets import load_dataset
 from tqdm.asyncio import tqdm_asyncio  # For progress bars in evaluate
 
-import wandb  # For wandb.Table
 from atroposlib.envs.base import (
     APIServerConfig,
     BaseEnv,