diff --git a/nodes/modifier_change/extrude_edges_mk2.py b/nodes/modifier_change/extrude_edges_mk2.py
index 36255050a36fa0a9a0946441f860f3056ab3c46e..a4b14f92a9a9752b93b30beadf042b3b648c750f 100644
--- a/nodes/modifier_change/extrude_edges_mk2.py
+++ b/nodes/modifier_change/extrude_edges_mk2.py
@@ -71,8 +71,13 @@ class SvExtrudeEdgesNodeMk2(bpy.types.Node, SverchCustomTreeNode, SvRecursiveNod
     def process_data(self, params):
 
         output_data = [[] for s in self.outputs]
-        extrude = extrude_edges if self.implementation == 'NUMPY' else extrude_edges_bmesh
+
         for vertices, edges, faces, edge_mask, face_data, matrices in zip(*params):
+
+            if edge_mask or self.implementation == 'BMESH':
+                extrude = extrude_edges_bmesh
+            else:
+                extrude = extrude_edges
             res = extrude(vertices, edges, faces, edge_mask, face_data, matrices)
             for o, r in zip(output_data, res):
                 o.append(r)