fix(archived): Fix for archived PageSections

flxlabs · Oct 24, 2018 · 1c82636 · 1c82636
1 parent 0f42fd9
commit 1c82636
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/PageSectionsExtension.php b/src/PageSectionsExtension.php
@@ -91,7 +91,7 @@ public function onAfterWrite()
 	{
 		parent::onAfterWrite();
 
-		if ($this->owner->ID && !$this->owner->__rewrite) {
+		if ($this->owner->ID && !$this->owner->__archived) {
 			$sections = $this->getPageSectionNames();
 
 			foreach ($sections as $sectionName) {
@@ -109,6 +109,7 @@ public function onBeforeArchive()
 	{
 		$sections = $this->getPageSectionNames();
 
+		$this->owner->__archived = true;
 		foreach ($sections as $sectionName) {
 			$name = "PageSection".$sectionName;
 			$this->owner->{$name . "ID"} = 0;