diff --git a/src/caustics/lenses/base.py b/src/caustics/lenses/base.py
index 4f51d0a1..803a1c2c 100644
--- a/src/caustics/lenses/base.py
+++ b/src/caustics/lenses/base.py
@@ -140,7 +140,7 @@ def forward_raytrace(
             number of random initialization points used to try and find image plane points.
         fov: float
             the field of view in which the initial random samples are taken.
-            *Unit: arcsec* 
+            *Unit: arcsec*
 
         Returns
         -------
diff --git a/src/caustics/lenses/mass_sheet.py b/src/caustics/lenses/mass_sheet.py
index e8a35a26..7e2d41c6 100644
--- a/src/caustics/lenses/mass_sheet.py
+++ b/src/caustics/lenses/mass_sheet.py
@@ -78,7 +78,7 @@ def reduced_deflection_angle(
         ----------
         x: Tensor
             x-coordinates in the lens plane.
-            *Unit: arcsec*            
+            *Unit: arcsec*
         y: Tensor
             y-coordinates in the lens plane.
             *Unit: arcsec*
diff --git a/src/caustics/lenses/multiplane.py b/src/caustics/lenses/multiplane.py
index 3478dfa1..7fc6c84a 100644
--- a/src/caustics/lenses/multiplane.py
+++ b/src/caustics/lenses/multiplane.py
@@ -238,10 +238,10 @@ def surface_density(
         ----------
         x: Tensor
             x-coordinates in the lens plane.
-            *Unit: arcsec*            
+            *Unit: arcsec*
         y: Tensor
             y-coordinates in the lens plane.
-            *Unit: arcsec*            
+            *Unit: arcsec*
         z_s: Tensor
             Redshifts of the sources.
         params: (Packed, optional)
@@ -251,7 +251,7 @@ def surface_density(
         -------
         Tensor
             Projected mass density [Solar Mass / Mpc^2].
-            *Unit: solMass / megaparsec^2* 
+            *Unit: solMass / megaparsec^2*
         Raises
         -------
         NotImplementedError