diff --git a/src/Extension/FluentCMSMainExtension.php b/src/Extension/FluentCMSMainExtension.php
index 3a7ef90f..318f13c2 100644
--- a/src/Extension/FluentCMSMainExtension.php
+++ b/src/Extension/FluentCMSMainExtension.php
@@ -106,7 +106,7 @@ public function publish_localised_copy($data, $form)
     protected function getRecordForLocalisedAction($data, $form): ?DataObject
     {
         $id = (int) $data['ID'];
-        $className = $this->owner->config()->get('tree_class');
+        $className = $this->owner->config()->get('model_class');
 
         if (!$id || !$className) {
             // Invalid inputs