diff --git a/pyproject.toml b/pyproject.toml
index 6b0e14cf..6a7f55f9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,11 +3,25 @@ requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools]
-packages = ["swiftsimio", "swiftsimio.initial_conditions", "swiftsimio.metadata", "swiftsimio.metadata.cosmology", "swiftsimio.metadata.metadata", "swiftsimio.metadata.particle", "swiftsimio.metadata.unit", "swiftsimio.metadata.writer", "swiftsimio.visualisation", "swiftsimio.visualisation.projection_backends", "swiftsimio.visualisation.slice_backends", "swiftsimio.visualisation.tools"]
+packages = [
+    "swiftsimio",
+    "swiftsimio.initial_conditions",
+    "swiftsimio.metadata",
+    "swiftsimio.metadata.cosmology",
+    "swiftsimio.metadata.metadata",
+    "swiftsimio.metadata.particle",
+    "swiftsimio.metadata.unit",
+    "swiftsimio.metadata.writer",
+    "swiftsimio.visualisation",
+    "swiftsimio.visualisation.projection_backends",
+    "swiftsimio.visualisation.slice_backends",
+    "swiftsimio.visualisation.tools",
+    "swiftsimio.visualisation.smoothing_length"
+]
 
 [project]
 name = "swiftsimio"
-version="8.0.0"
+version="8.0.1"
 authors = [
     { name="Josh Borrow", email="josh@joshborrow.com" },
 ]
diff --git a/swiftsimio/__version__.py b/swiftsimio/__version__.py
index 73d4c8be..9e17e009 100644
--- a/swiftsimio/__version__.py
+++ b/swiftsimio/__version__.py
@@ -1 +1 @@
-__version__ = "8.0.0"
+__version__ = "8.0.1"
diff --git a/swiftsimio/visualisation/__init__.py b/swiftsimio/visualisation/__init__.py
index 05928f1b..b8bc3b34 100644
--- a/swiftsimio/visualisation/__init__.py
+++ b/swiftsimio/visualisation/__init__.py
@@ -6,3 +6,4 @@
 from .slice import slice_scatter as slice
 from .slice import slice_gas, slice_gas_pixel_grid
 from .smoothing_length import generate_smoothing_lengths
+
diff --git a/swiftsimio/visualisation/smoothing_length/__init__.py b/swiftsimio/visualisation/smoothing_length/__init__.py
index ee97dd06..75ffd81f 100644
--- a/swiftsimio/visualisation/smoothing_length/__init__.py
+++ b/swiftsimio/visualisation/smoothing_length/__init__.py
@@ -1,12 +1,11 @@
-from swiftsimio.visualisation.smoothing_length.sph import get_hsml as sph_get_hsml
-from swiftsimio.visualisation.smoothing_length.nearest_neighbours import (
+from .sph import get_hsml as sph_get_hsml
+from .nearest_neighbours import (
     get_hsml as nearest_neighbours_get_hsml,
 )
-from swiftsimio.visualisation.smoothing_length.generate import (
+from .generate import (
     generate_smoothing_lengths,
 )
 
-
 backends_get_hsml = {
     "sph": sph_get_hsml,
     "nearest_neighbours": nearest_neighbours_get_hsml,