diff --git a/source/tests/pt/model/test_linear_atomic_model_stat.py b/source/tests/pt/model/test_linear_atomic_model_stat.py
index f3fe4f63d4..490d777cb1 100644
--- a/source/tests/pt/model/test_linear_atomic_model_stat.py
+++ b/source/tests/pt/model/test_linear_atomic_model_stat.py
@@ -206,8 +206,8 @@ def test_linear_atomic_model_stat_with_bias(self):
                 [4.0, 5.0, 6.0],
                 [7.0, 8.0, 9.0],
             ]
-        ).reshape(nf, nloc,  *linear_model.fitting_output_def()["energy"].shape)
-        
+        ).reshape(nf, nloc, *linear_model.fitting_output_def()["energy"].shape)
+
         np.testing.assert_almost_equal(ret0, expected_ret0)
 
         # 2. test bias is applied