diff --git a/source/tests/tf/test_fitting_dos.py b/source/tests/tf/test_fitting_dos.py
index bcf3b8368e..f9df5fc126 100644
--- a/source/tests/tf/test_fitting_dos.py
+++ b/source/tests/tf/test_fitting_dos.py
@@ -189,12 +189,20 @@ def test_fitting(self):
         pred_atom_dos = pred_atom_dos.reshape(-1, numb_dos)
 
         ref_atom_dos_1 = [
-            -0.32495014, -0.32495882, -0.32496842, -0.32495892, -0.32495469,
-       -0.32496075
+            -0.32495014,
+            -0.32495882,
+            -0.32496842,
+            -0.32495892,
+            -0.32495469,
+            -0.32496075,
         ]
         ref_atom_dos_2 = [
-            0.21549911, 0.21550413, 0.21551077, 0.21550547, 0.21550303,
-       0.21550645
+            0.21549911,
+            0.21550413,
+            0.21551077,
+            0.21550547,
+            0.21550303,
+            0.21550645,
         ]
         places = 4
         np.testing.assert_almost_equal(pred_atom_dos[:, 0], ref_atom_dos_1, places)