1 files changed, 1 insertions, 5 deletions
diff --git a/modules/organize/controllers/organize.php b/modules/organize/controllers/organize.php
index cfd385fe..82f74c9e 100644
--- a/modules/organize/controllers/organize.php
+++ b/modules/organize/controllers/organize.php
@@ -169,6 +169,7 @@ class Organize_Controller extends Controller {
         "editable" => false,
         "expandable" => false,
         "id" => $child->id,
+        "leaf" => $child->children_count(array(array("type", "=", "album"))) == 0,
         "text" => $child->title,
         "nodeType" => "async");
 
@@ -177,11 +178,6 @@ class Organize_Controller extends Controller {
         $node["children"] = $this->_get_tree($child, $selected);
         $node["expanded"] = true;
       }
-
-      if ($child->children_count(array(array("type", "=", "album"))) == 0) {
-        $node["children"] = array();
-        $node["expanded"] = true;
-      }
       $tree[] = $node;
     }
     return $tree;