From ff379f5ebf81a7252919f577b2a979d9b6169fa3 Mon Sep 17 00:00:00 2001
From: Sam Georges <sam@daftspunk.com>
Date: Fri, 13 Jun 2014 18:34:54 +1000
Subject: [PATCH] Switch to new Nested Set model trait -
 October\Rain\Database\Traits\NestedTree

---
 modules/backend/formwidgets/Relation.php | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/backend/formwidgets/Relation.php b/modules/backend/formwidgets/Relation.php
index edb9bdb8b..b1e4e204b 100644
--- a/modules/backend/formwidgets/Relation.php
+++ b/modules/backend/formwidgets/Relation.php
@@ -108,7 +108,7 @@ class Relation extends FormWidgetBase
             $query->where($relatedObj->getKeyName(), '<>', $this->model->id);
          }
 
-         if ($relatedObj->isClassExtendedWith('October.Rain.Database.Behaviors.NestedSetModel'))
+         if (in_array('October\Rain\Database\Traits\NestedTree', class_uses($relatedObj)))
             $field->options = $query->listsNested($this->nameColumn, $relatedObj->getKeyName());
          else
             $field->options = $query->lists($this->nameColumn, $relatedObj->getKeyName());