diff --git a/codes/tb/test_tb.py b/codes/tb/test_tb.py
index cb339958ab5c72917f173bf8dd5af64cd6737efa..a4d442bae6152d2d2b661e399f6e7b670b54106c 100644
--- a/codes/tb/test_tb.py
+++ b/codes/tb/test_tb.py
@@ -1,7 +1,7 @@
 # %%
 import numpy as np
 from codes.tb.tb import compareDicts
-from codes.kwant_helper import utils    
+from codes.kwant_helper import utils
 import itertools as it
 from codes.tb.transforms import kfunc2tb, tb2kfunc, tb2kham, tb2khamvector
 import pytest
@@ -24,10 +24,21 @@ def test_fourier():
     tb_new = kfunc2tb(kfunc, nK, ndim=ndim)
     compareDicts(h_0, tb_new)
 
-@pytest.mark.repeat(repeatNumber):
-def test_tbkham_transform(): 
-    vectors = ((0, 0), (1, 0), (-1, 0), (0, 1), (0, -1), (1, -1), (-1, 1), (1, 1), (-1, -1))
+
+@pytest.mark.repeat(repeatNumber)
+def test_tbkham_transform():
+    vectors = (
+        (0, 0),
+        (1, 0),
+        (-1, 0),
+        (0, 1),
+        (0, -1),
+        (1, -1),
+        (-1, 1),
+        (1, 1),
+        (-1, -1),
+    )
     ndof = 10
     h_0 = utils.generate_guess(vectors, ndof)
 
-    assert np.allclose(tb2kham(h_0, nK=nK, ndim=2), tb2khamvector(h_0, nK=nK, ndim=2))
\ No newline at end of file
+    assert np.allclose(tb2kham(h_0, nK=nK, ndim=2), tb2khamvector(h_0, nK=nK, ndim=2))