diff --git a/src/CampaignAdmin.php b/src/CampaignAdmin.php
index 5402113..51465c6 100644
--- a/src/CampaignAdmin.php
+++ b/src/CampaignAdmin.php
@@ -61,7 +61,7 @@ class CampaignAdmin extends LeftAndMain implements PermissionProvider
 
     private static $menu_icon_class = 'font-icon-page-multiple';
 
-    private static $tree_class = ChangeSet::class;
+    private static $model_class = ChangeSet::class;
 
     /**
      * Show published changesets
@@ -144,7 +144,7 @@ public function getClientConfig(): array
                 'url' => $this->Link('removeCampaignItem/:id/:itemId'),
                 'method' => 'post'
             ],
-            'treeClass' => $this->config()->get('tree_class')
+            'treeClass' => $this->config()->get('model_class')
         ]);
     }
 
@@ -235,7 +235,7 @@ protected function getListResource()
         $items = $this->getListItems();
         $count = $items->count();
         /** @var string $treeClass */
-        $treeClass = $this->config()->get('tree_class');
+        $treeClass = $this->config()->get('model_class');
         $hal = [
             'count' => $count,
             'total' => $count,
diff --git a/src/CampaignAdminList.php b/src/CampaignAdminList.php
index 3bc780a..aeab18f 100644
--- a/src/CampaignAdminList.php
+++ b/src/CampaignAdminList.php
@@ -32,7 +32,7 @@ public function getSchemaDataDefaults()
 
         // Get endpoints from admin
         $admin = CampaignAdmin::singleton();
-        $data['data']['recordType'] = $admin->config()->uninherited('tree_class');
+        $data['data']['recordType'] = $admin->config()->uninherited('model_class');
         $oneSetAction = $admin->Link("set") . "/:id";
         $setsAction = $admin->Link("sets");
         $schemaEndpoint = $admin->Link("schema") . "/DetailEditForm";