diff --git a/reasoning_gym/algebra/__init__.py b/reasoning_gym/algebra/__init__.py
index 28cfe7a2..84c3d302 100644
--- a/reasoning_gym/algebra/__init__.py
+++ b/reasoning_gym/algebra/__init__.py
@@ -1,5 +1,5 @@
-from .simple_equations import SimpleEquationsConfig, SimpleEquationsDataset, simple_equations_dataset
 from .polynomial_equations import PolynomialEquationsConfig, PolynomialEquationsDataset, polynomial_equations_dataset
+from .simple_equations import SimpleEquationsConfig, SimpleEquationsDataset, simple_equations_dataset
 
 __all__ = [
     "SimpleEquationsDataset",
diff --git a/reasoning_gym/algebra/polynomial_equations.py b/reasoning_gym/algebra/polynomial_equations.py
index ac02c19c..60935013 100644
--- a/reasoning_gym/algebra/polynomial_equations.py
+++ b/reasoning_gym/algebra/polynomial_equations.py
@@ -1,10 +1,10 @@
 import random
 import string
 from dataclasses import dataclass
-from typing import Optional, Tuple, List
+from typing import List, Optional, Tuple
 
 import sympy
-from sympy import Symbol, Eq, solve, expand
+from sympy import Eq, Symbol, expand, solve
 
 from ..dataset import ProceduralDataset
 
diff --git a/reasoning_gym/games/__init__.py b/reasoning_gym/games/__init__.py
index 5b9dfa8c..507fc7d9 100644
--- a/reasoning_gym/games/__init__.py
+++ b/reasoning_gym/games/__init__.py
@@ -5,9 +5,9 @@ Game tasks for training reasoning capabilities:
 - Strategy games
 """
 
+from .maze import MazeConfig, MazeDataset, maze_dataset
 from .mini_sudoku import MiniSudokuConfig, MiniSudokuDataset, mini_sudoku_dataset
 from .sudoku import SudokuConfig, SudokuDataset, sudoku_dataset
-from .maze import MazeConfig, MazeDataset, maze_dataset
 
 __all__ = [
     "MiniSudokuConfig",
diff --git a/reasoning_gym/games/maze.py b/reasoning_gym/games/maze.py
index 69ac43c1..439c2809 100644
--- a/reasoning_gym/games/maze.py
+++ b/reasoning_gym/games/maze.py
@@ -3,7 +3,6 @@ import string
 from dataclasses import dataclass
 from typing import Optional
 
-
 from ..dataset import ProceduralDataset
 
 
diff --git a/tests/test_polynomial_equations.py b/tests/test_polynomial_equations.py
index daa33e2e..16186092 100644
--- a/tests/test_polynomial_equations.py
+++ b/tests/test_polynomial_equations.py
@@ -1,5 +1,5 @@
 import pytest
-from sympy import sympify, Symbol
+from sympy import Symbol, sympify
 
 from reasoning_gym.algebra.polynomial_equations import (
     PolynomialEquationsConfig,