diff --git a/freecad/kiconnect/bases/BaseObject.py b/freecad/kiconnect/bases/BaseObject.py
index c503004..62bf206 100644
--- a/freecad/kiconnect/bases/BaseObject.py
+++ b/freecad/kiconnect/bases/BaseObject.py
@@ -26,6 +26,9 @@ class BaseObject:
     def onChanged(self, feature, prop):
         pass
 
+    def onDocumentRestored(self, feature):
+        self.feature = feature
+
     def setup_extensions(self, feature):
         if hasattr(self, 'EXTENSIONS'):
             for ext in self.EXTENSIONS: