diff --git a/reasoning_gym/algebra/complex_arithmetic.py b/reasoning_gym/algebra/complex_arithmetic.py
index 80124447..7c749eaa 100644
--- a/reasoning_gym/algebra/complex_arithmetic.py
+++ b/reasoning_gym/algebra/complex_arithmetic.py
@@ -72,7 +72,7 @@ class ComplexArithmeticDataset(ProceduralDataset):
             # For other operations, generate numbers normally
             a = self._generate_complex(rng)
             b = self._generate_complex(rng)
-            
+
             # Calculate result
             if op == "+":
                 result = a + b
diff --git a/tests/test_tsumego.py b/tests/test_tsumego.py
index 86ac203f..c1d8d7b6 100644
--- a/tests/test_tsumego.py
+++ b/tests/test_tsumego.py
@@ -1,8 +1,9 @@
 """Tests for Ttsumego problem generation"""
 
-import pytest
 from random import Random
 
+import pytest
+
 from reasoning_gym.games.tsumego import TsumegoConfig, TsumegoDataset