diff --git a/src/SMESH_SWIG/StdMeshersBuilder.py b/src/SMESH_SWIG/StdMeshersBuilder.py
index 07fbc4d51..e2f1d5f46 100644
--- a/src/SMESH_SWIG/StdMeshersBuilder.py
+++ b/src/SMESH_SWIG/StdMeshersBuilder.py
@@ -302,8 +302,8 @@ class StdMeshersBuilder_Segment(Mesh_Algorithm):
         store_geom = self.geom
         if type(vertex) is types.IntType:
             if vertex == 0 or vertex == 1:
-                from geompy import ShapeType
-                vertex = self.mesh.geompyD.ExtractShapes(self.geom, ShapeType["VERTEX"],True)[vertex]
+                from salome.geom import geomBuilder
+                vertex = self.mesh.geompyD.ExtractShapes(self.geom, geomBuilder.geomBuilder.ShapeType["VERTEX"],True)[vertex]
                 self.geom = vertex
                 pass
             pass