diff --git a/ocf_datapipes/visualisation/__init__.py b/ocf_datapipes/visualisation/__init__.py
index d654617a..bbc352f0 100644
--- a/ocf_datapipes/visualisation/__init__.py
+++ b/ocf_datapipes/visualisation/__init__.py
@@ -1,3 +1,3 @@
 """Tools for plotting and visualizing data."""
 
-from .batch import visualise_batch
\ No newline at end of file
+from .batch import visualise_batch