diff --git a/dianna/visualization/__init__.py b/dianna/visualization/__init__.py
index a9dd0aac..0e08ae8c 100644
--- a/dianna/visualization/__init__.py
+++ b/dianna/visualization/__init__.py
@@ -1,6 +1,6 @@
 # flake8: noqa: F401
 """Tools for visualization of model explanations."""
 from .image import plot_image
-from .text import highlight_text
 from .tabular import plot_tabular
+from .text import highlight_text
 from .timeseries import plot_timeseries