diff --git a/DeepSDFStruct/SDF.py b/DeepSDFStruct/SDF.py
index e9f0bec..00d6d67 100644
--- a/DeepSDFStruct/SDF.py
+++ b/DeepSDFStruct/SDF.py
@@ -1151,7 +1151,7 @@ def __init__(self, line_mesh: gustaf.Edges, thickness, smoothness=0):
         self.geometric_dim = line_mesh.vertices.shape[1]
 
     def _get_domain_bounds(self):
-        return self.line_mesh.bounds()
+        return torch.tensor(self.line_mesh.bounds())
 
     def _set_param(self, parameters):
         self.t = parameters[0]