diff --git a/src/ansys/geometry/core/plotting/plotter.py b/src/ansys/geometry/core/plotting/plotter.py
index 18ce273c39..00de4bfb00 100644
--- a/src/ansys/geometry/core/plotting/plotter.py
+++ b/src/ansys/geometry/core/plotting/plotter.py
@@ -402,6 +402,8 @@ def add_design_point(self, design_point: DesignPoint, **plotting_options) -> Non
         actor = self.scene.add_mesh(design_point._to_polydata(), **plotting_options)
 
         # save the actor to the object/actor map
+        design_point = GeomObjectPlot(actor=actor, object=design_point, add_body_edges=False)
+
         self._geom_object_actors_map[actor] = design_point
 
     def add(
diff --git a/src/ansys/geometry/core/plotting/plotter_helper.py b/src/ansys/geometry/core/plotting/plotter_helper.py
index d2db663066..f93c486430 100644
--- a/src/ansys/geometry/core/plotting/plotter_helper.py
+++ b/src/ansys/geometry/core/plotting/plotter_helper.py
@@ -46,6 +46,8 @@
     ViewButton,
     ViewDirection,
 )
+from ansys.geometry.core.sketch.face import SketchFace
+from ansys.geometry.core.sketch.sketch import Sketch
 
 
 class PlotterHelper:
@@ -229,10 +231,13 @@ def compute_edge_object_map(self) -> Dict[pv.Actor, EdgePlot]:
         """
         for object in self._geom_object_actors_map.values():
             # get edges only from bodies
+            geom_obj = object.object
             if (
-                isinstance(object, Body)
-                or isinstance(object, MasterBody)
-                or isinstance(object, Face)
+                isinstance(geom_obj, Body)
+                or isinstance(geom_obj, MasterBody)
+                or isinstance(geom_obj, Face)
+                or isinstance(geom_obj, SketchFace)
+                or isinstance(geom_obj, Sketch)
             ):
                 for edge in object.edges:
                     self._edge_actors_map[edge.actor] = edge