fix(page-section): Fix parent changing class

flxlabs · Mar 18, 2019 · 1b86bba · 1b86bba
1 parent 3f85484
commit 1b86bba
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/PageSectionsExtension.php b/src/PageSectionsExtension.php
@@ -100,6 +100,9 @@ public function onAfterWrite()
 				if (!$this->owner->{$name . "ID"}) {
 					// Restore or create a page section if we don't have one yet
 					$this->restoreOrCreate($sectionName);
+				} else if ($this->owner->$name()->__ParentClass !== $this->owner->ClassName) {
+					$this->owner->$name()->__ParentClass = $this->owner->ClassName;
+					$this->owner->$name()->write();
 				}
 			}
 		}