fixed a typo

MintonGroup · Nov 15, 2023 · 137e73b · 137e73b
1 parent f52809e
commit 137e73b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/swiftest/sph_harmonics.py b/swiftest/sph_harmonics.py
@@ -83,7 +83,8 @@ def clm_from_ellipsoid(mass, density, a, b = None, c = None, lmax = 6, lref_radi
             ref_radius = shape_SH.expand(normalization = '4pi').coeffs[0, 0, 0]
             return clm, ref_radius
         elif(lref_radius == True and ref_radius is not None):
-            clm = clm.change_ref(r0 = ref_radius)
+            clm_class = clm_class.change_ref(r0 = ref_radius)
+            clm = clm_class.to_array(normalization = '4pi')
             return clm, ref_radius
         else:
             return clm
@@ -140,7 +141,8 @@ def clm_from_relief(mass, density, grid, lmax = 6, lref_radius = False, ref_radi
             ref_radius = shape_SH.expand(normalization = '4pi').coeffs[0, 0, 0]
             return clm, ref_radius
         elif(lref_radius == True and ref_radius is not None):
-            clm = clm.change_ref(r0 = ref_radius)
+            clm_class = clm_class.change_ref(r0 = ref_radius)
+            clm = clm_class.to_array(normalization = '4pi')
             return clm, ref_radius
         else:
             return clm