diff --git a/reasoning_gym/logic/__init__.py b/reasoning_gym/logic/__init__.py
index f3dad81c..a0d83f44 100644
--- a/reasoning_gym/logic/__init__.py
+++ b/reasoning_gym/logic/__init__.py
@@ -3,11 +3,11 @@ Logic tasks for training reasoning capabilities.
 """
 
 from .aiw import AliceInWonderlandConfig, AliceInWonderlandDataset
+from .circuit_logic import CircuitLogicConfig, CircuitLogicDataset
 from .propositional_logic import PropositionalLogicConfig, PropositionalLogicDataset
 from .self_reference import SelfReferenceConfig, SelfReferenceDataset
 from .syllogisms import SyllogismConfig, SyllogismDataset, Term
 from .zebra_puzzles import ZebraConfig, ZebraDataset
-from .circuit_logic import CircuitLogicConfig, CircuitLogicDataset
 
 __all__ = [
     "AliceInWonderlandConfig",
diff --git a/reasoning_gym/logic/circuit_logic.py b/reasoning_gym/logic/circuit_logic.py
index 9d46f82f..bcc2e4c6 100644
--- a/reasoning_gym/logic/circuit_logic.py
+++ b/reasoning_gym/logic/circuit_logic.py
@@ -1,6 +1,7 @@
 from dataclasses import dataclass
-from typing import Any, Dict, List, Optional, Tuple
 from random import Random
+from typing import Any, Dict, List, Optional, Tuple
+
 from ..factory import ProceduralDataset, register_dataset
 
 VERT = "│"