diff --git a/src/Extensions/SiteTreeExtension.php b/src/Extensions/SiteTreeExtension.php index 94fd690..4b26d2a 100755 --- a/src/Extensions/SiteTreeExtension.php +++ b/src/Extensions/SiteTreeExtension.php @@ -44,8 +44,20 @@ class SiteTreeExtension extends DataExtension public function ShowSidebar() { $obj = $this->owner; + $area = $obj->ElementalArea(); + if(!$area) { + return true; + } + $els = $area->Elements(); + if(!$els) { + return true; + } + $els = $els->find('ClassName', SidebarElement::class); + if(!$els) { + return true; + } - if ($obj->ElementalArea()->Elements()->find('ClassName', SidebarElement::class)->first()) { + if ($els->first()) { return false; }