diff --git a/workflows/domino_design_sensitivities/design_datapipe.py b/workflows/domino_design_sensitivities/design_datapipe.py
index 8d0f12f..16d0a1b 100644
--- a/workflows/domino_design_sensitivities/design_datapipe.py
+++ b/workflows/domino_design_sensitivities/design_datapipe.py
@@ -94,7 +94,14 @@ def __init__(
 
         surface_normals = -1.0 * np.array(mesh.cell_normals, dtype=np.float32)
 
-        center_of_mass = calculate_center_of_mass(stl_centers, surface_areas)
+        center_of_mass = (
+            calculate_center_of_mass(
+                torch.as_tensor(stl_centers), torch.as_tensor(surface_areas)
+            )
+            .detach()
+            .cpu()
+            .numpy()
+        )
 
         s_max = np.asarray(bounding_box_surface[1])
         s_min = np.asarray(bounding_box_surface[0])