FIX Use Updated names of top page extensions

silverstripe · Sep 4, 2024 · 2a7287f · 2a7287f
1 parent 7c53d75
commit 2a7287f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/Extensions/UsedOnTableExtension.php b/src/Extensions/UsedOnTableExtension.php
@@ -3,7 +3,8 @@
 namespace SilverStripe\LinkField\Extensions;
 
 use DNADesign\Elemental\Models\BaseElement;
-use DNADesign\Elemental\TopPage\DataExtension as TopPageExtension;
+use DNADesign\Elemental\Extensions\TopPageElementExtension;
+use DNADesign\Elemental\Extensions\TopPageFluentElementExtension;
 use SilverStripe\Core\Extension;
 use SilverStripe\LinkField\Models\FileLink;
 use SilverStripe\ORM\DataObject;
@@ -23,7 +24,11 @@ protected function updateUsageAncestorDataObjects(array &$ancestorDataObjects, D
         if (!class_exists(BaseElement::class) || !is_a($owner, BaseElement::class)) {
             return;
         }
-        $page = $owner->hasExtension(TopPageExtension::class) ? $owner->getTopPage() : $owner->getPage();
+        if ($owner->hasExtension(TopPageElementExtension::class)) {
+            $page = $owner->getTopPage();
+        } else {
+            $page = $owner->getPage();
+        }
         if (!$page?->exists()) {
             return;
         }