diff --git a/brainglobe_heatmap/heatmaps.py b/brainglobe_heatmap/heatmaps.py
index a7e7b2a..1bd21f0 100644
--- a/brainglobe_heatmap/heatmaps.py
+++ b/brainglobe_heatmap/heatmaps.py
@@ -252,4 +252,4 @@ def plot(
             ax.legend()
         plt.show()
 
-        return f
\ No newline at end of file
+        return f