diff --git a/src/SMESH/SMESH_Gen.cxx b/src/SMESH/SMESH_Gen.cxx
index f61acd5f9..f2e4e0d9f 100644
--- a/src/SMESH/SMESH_Gen.cxx
+++ b/src/SMESH/SMESH_Gen.cxx
@@ -184,9 +184,9 @@ const std::function<void(int,
   if (sm->GetComputeState() == SMESH_subMesh::READY_TO_COMPUTE)
   {
     sm->SetAllowedSubShapes( fillAllowed( shapeSM, aShapeOnly, allowedSubShapes ));
-    //setCurrentSubMesh( sm );
+    setCurrentSubMesh( sm );
     sm->ComputeStateEngine(event);
-    //setCurrentSubMesh( nullptr );
+    setCurrentSubMesh( nullptr );
     sm->SetAllowedSubShapes( nullptr );
   }