- aPart->data()->docRef(ModelAPI_ResultPart::DOC_REF())->value()->close();
- aRoot->removeFeature(aFeature);
+ aPart->data()->document(ModelAPI_ResultPart::DOC_REF())->value()->close();
+ std::set<std::shared_ptr<ModelAPI_Feature> > aRefFeatures;
+ aRoot->refsToFeature(aFeature, aRefFeatures);
+ if (aRefFeatures.empty())
+ aRoot->removeFeature(aFeature);