diff --git a/heat/__init__.py b/heat/__init__.py
index 18a6399..90e390a 100644
--- a/heat/__init__.py
+++ b/heat/__init__.py
@@ -2,6 +2,7 @@
 
 from ._version import __version__
 from .bmi_heat import BmiHeat
-from .heat import Heat, solve_2d
+from .heat import Heat
+from .heat import solve_2d
 
 __all__ = ["__version__", "BmiHeat", "solve_2d", "Heat"]
diff --git a/tests/get_value_test.py b/tests/get_value_test.py
index 117525a..de265a2 100644
--- a/tests/get_value_test.py
+++ b/tests/get_value_test.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 import numpy as np
-from numpy.testing import assert_array_almost_equal, assert_array_less
+from numpy.testing import assert_array_almost_equal
+from numpy.testing import assert_array_less
 
 from heat import BmiHeat
 
diff --git a/tests/irf_test.py b/tests/irf_test.py
index bfe2dc3..869309f 100644
--- a/tests/irf_test.py
+++ b/tests/irf_test.py
@@ -3,7 +3,9 @@
 
 import numpy as np
 import yaml
-from numpy.testing import assert_almost_equal, assert_array_equal, assert_array_less
+from numpy.testing import assert_almost_equal
+from numpy.testing import assert_array_equal
+from numpy.testing import assert_array_less
 
 from heat import BmiHeat