diff --git a/product/ERP5CPS/CPSCorePatch.py b/product/ERP5CPS/CPSCorePatch.py
index ac9f3be9f4962eabe52302095b374b42419fed4b..1c6de02d0001299f2d03e5a3929d9ab6a89266d2 100755
--- a/product/ERP5CPS/CPSCorePatch.py
+++ b/product/ERP5CPS/CPSCorePatch.py
@@ -25,6 +25,7 @@ from Products.ERP5Type.Document.Folder import Folder
 from Products.CMFCore.CMFCorePermissions import View
 from Products.CMFCore.CMFCorePermissions import ModifyPortalContent
 from Products.CMFCore.CMFCorePermissions import ViewManagementScreens
+from Products.CMFCore.utils import getToolByName
 from zLOG import LOG
 
 # First we should make ProxyBase a subclass of Base
diff --git a/product/ERP5CPS/CPSDocumentPatch.py b/product/ERP5CPS/CPSDocumentPatch.py
index e4b84cc0e239924cb7f0465414800a11ea6ef3dd..1f3d6b8406e5bedb8e294df3e85e314fb72b793b 100755
--- a/product/ERP5CPS/CPSDocumentPatch.py
+++ b/product/ERP5CPS/CPSDocumentPatch.py
@@ -95,4 +95,6 @@ class PatchedCPSDocument(CPSDocument):
 CPSDocument.getProperty = PatchedCPSDocument.getProperty
 CPSDocument._propertyMap = PatchedCPSDocument._propertyMap
 CPSDocument.setProperty = Base.setProperty
+CPSDocument._setProperty = Base._setProperty
 CPSDocument.asXML = Base.asXML
+CPSDocument._edit = Base._edit