diff --git a/meanfi/tests/test_kwant.py b/meanfi/tests/test_kwant.py
index 48cb11709b3a80f6b046ecb3ec9c76175f1f6656..3785f0071eb1ad414966e383ae6120db925b56e6 100644
--- a/meanfi/tests/test_kwant.py
+++ b/meanfi/tests/test_kwant.py
@@ -42,7 +42,7 @@ def test_kwant_conversion(seed):
         random_tb, list(dummy_tb.sites()), dummy_tb.symmetry.periods
     )
     # convert builder back to tb and compare
-    random_builder_tb = builder_to_tb(random_builder, return_data=True)
+    random_builder_tb = builder_to_tb(random_builder)
     compare_dicts(random_tb, random_builder_tb)
 
 
@@ -107,16 +107,20 @@ def test_kwant_supercell(seed):
         site1, site2 = site_pair[0]
         if site1 == site2:
             assert np.isclose(
-                random_builder[site1](site=site1, alpha=params["alpha"], beta=params["beta"]),
+                random_builder[site1](
+                    site=site1, alpha=params["alpha"], beta=params["beta"]
+                ),
                 random_builder_test[site1],
             ).all()
         else:
             try:
                 assert np.isclose(
-                    random_builder[site1, site2](site1, site2, gamma=params["gamma"], delta=params["delta"]),
+                    random_builder[site1, site2](
+                        site1, site2, gamma=params["gamma"], delta=params["delta"]
+                    ),
                     random_builder_test[site1, site2],
                 ).all()
             except KeyError:
                 continue
             except:
-                raise
\ No newline at end of file
+                raise