diff --git a/src/laue_dials/algorithms/laue.py b/src/laue_dials/algorithms/laue.py
index 7576ae1..a2207a4 100644
--- a/src/laue_dials/algorithms/laue.py
+++ b/src/laue_dials/algorithms/laue.py
@@ -86,9 +86,8 @@ def gen_beam_models(expts, refls):
     # Imports
     from copy import deepcopy
 
-    from dials.algorithms.refinement.prediction.managed_predictors import (
-        ExperimentsPredictorFactory,
-    )
+    from dials.algorithms.refinement.prediction.managed_predictors import \
+        ExperimentsPredictorFactory
 
     # Instantiate new ExperimentList/reflection_table
     new_expts = ExperimentList()
diff --git a/tests/algorithms/test_laue.py b/tests/algorithms/test_laue.py
index 3a94041..8992a11 100644
--- a/tests/algorithms/test_laue.py
+++ b/tests/algorithms/test_laue.py
@@ -1,15 +1,11 @@
 import numpy as np
 import pytest
-from dxtbx.model import ExperimentList
 from dials.array_family.flex import reflection_table
-from laue_dials.algorithms.laue import (
-    store_wavelengths,
-    remove_beam_models,
-    gen_beam_models,
-    LaueBase,
-    LaueAssigner,
-    LauePredictor,
-)
+from dxtbx.model import ExperimentList
+
+from laue_dials.algorithms.laue import (LaueAssigner, LaueBase, LauePredictor,
+                                        gen_beam_models, remove_beam_models,
+                                        store_wavelengths)
 
 
 @pytest.fixture