diff --git a/gui.py b/gui.py
index e5e3c26..caf3504 100644
--- a/gui.py
+++ b/gui.py
@@ -37,6 +37,9 @@ class SelectionObserver:
             FreeCADGui.Selection.updateSelection(vis,obj,subname)
 
     def setElementVisible(self,docname,objname,subname,vis,presel=False):
+        if FreeCAD.isRestoring():
+            self.resetElementVisible()
+            return
         if not AsmCmdManager.AutoElementVis:
             self.elements.clear()
             return