diff --git a/src/geomtools/sfx/__init__.py b/src/geomtools/sfx/__init__.py
index ae8792d9a824bfb5d38fb06395a6876bdfd945aa..e58dfa85da4b503e77d566af72767bf8649a076c 100644
--- a/src/geomtools/sfx/__init__.py
+++ b/src/geomtools/sfx/__init__.py
@@ -1,19 +1,12 @@
 # flake8: noqa E401
-from .crystfelio import (
-    parse_crystfel_streamfile, read_crystfel_streamfile, extract_geometry,
-    extract_cell,
-)
-from .draw import (
-    plot_center_shift, plot_cell_parameters, plot_peakogram, plot_powder,
-    plot_geoptimiser_errormap,
-)
-from .lattice import (
-    cell_volume, spacing, ph_en_to_lambda, get_q_from_xyz,
-    get_min_bragg_dist, read_crystfel_cell,
-)
-from .misc import (
-    gauss2d_fit, ellipse, get_peak_position, avg_pixel_displacement,
-    rmsd_per_group,
-)
-from .xwizio import parse_xwiz_summary
+from .crystfelio import (extract_cell, extract_geometry,
+                         parse_crystfel_streamfile, read_crystfel_streamfile)
+from .draw import (plot_cell_parameters, plot_center_shift,
+                   plot_geoptimiser_errormap, plot_peakogram, plot_powder)
+from .lattice import (cell_volume, get_min_bragg_dist, get_q_from_xyz,
+                      ph_en_to_lambda, read_crystfel_cell, spacing)
+from .misc import (avg_pixel_displacement, badpixels_mask, ellipse,
+                   gauss2d_fit, get_peak_position, pixels_to_image,
+                   rmsd_per_group)
 from .refine import refine_geometry
+from .xwizio import parse_xwiz_summary