diff --git a/application/controllers/projects.php b/application/controllers/projects.php
index af91835..e81c329 100644
--- a/application/controllers/projects.php
+++ b/application/controllers/projects.php
@@ -6,19 +6,23 @@ class Projects extends CI_Controller {
 	 * List all projects the user has access to.
 	 */
 	public function getAvailable() {
-		$this->load->model('Project');
+		$this->load->model('project');
+		$this->load->helper('tree');
 
 		$path = $this->input->get_post('node');
 
 		switch($path) {
 			case '/projects/own':
-				$projects = $this->Project->getOwn();
+				$projects = $this->project->getOwn();
+				array_walk($projects, 'set_tree_icons', base_url() . 'assets/images/icons/document.png');
 				break;
 			case '/projects/shared':
-				$projects = $this->Project->getShared();
+				$projects = $this->project->getShared();
+				array_walk($projects, 'set_tree_icons', base_url() . 'assets/images/icons/document.png');
 				break;
 			case '/projects/public':
-				$projects = $this->Project->getPublic();
+				$projects = $this->project->getPublic();
+				array_walk($projects, 'set_tree_icons', base_url() . 'assets/images/icons/document.png');
 				break;
 			default:
 				$projects = array(
diff --git a/application/helpers/tree_helper.php b/application/helpers/tree_helper.php
new file mode 100644
index 0000000..f28808d
--- /dev/null
+++ b/application/helpers/tree_helper.php
@@ -0,0 +1,9 @@
+<?php defined('BASEPATH') || exit("No direct script access allowed");
+
+if (!function_exists('set_tree_icons')) {
+    function set_tree_icons(&$value, $key, $icon) {
+        if (isset($value['text']) && !isset($value['icon'])) {
+            $value['icon'] = $icon;
+        }
+    }
+}
\ No newline at end of file
diff --git a/application/models/project.php b/application/models/project.php
index d8e48c7..e25a2ef 100644
--- a/application/models/project.php
+++ b/application/models/project.php
@@ -1,7 +1,7 @@
 <?php
 
 class Project extends CI_Model {
-	
+
 	public function getOwn() {
 		$query = $this->db->where(array('owner' => '215cd70f310ae6ae'))
 				->order_by('lastaccess', 'desc')
@@ -15,18 +15,17 @@ class Project extends CI_Model {
 			$ownProjects[$i]['cls'] = 'folder';
 			$ownProjects[$i]['text'] = $project['name'];
 			$ownProjects[$i]['leaf'] = true;
-			$ownProjects[$i]['icon'] = "/ScattPort/assets/images/icons/document.png";
 			$i++;
 		}
-		
+
 		return $ownProjects;
 	}
-	
+
 	public function getShared() {
 		$this->db->select('*')->from('shares')->order_by('lastaccess', 'desc')->where(array('user_id' => '215cd70f310ae6ae'));
 		$this->db->join('projects', 'projects.id = shares.project_id');
 		$query = $this->db->get();
-		
+
 		$projects = $query->result_array();
 		$sharedCount = $query->num_rows();
 
@@ -36,13 +35,12 @@ class Project extends CI_Model {
 			$sharedProjects[$i]['cls'] = 'folder';
 			$sharedProjects[$i]['text'] = $project['name'];
 			$sharedProjects[$i]['leaf'] = true;
-			$sharedProjects[$i]['icon'] = "/ScattPort/assets/images/icons/document.png";
 			$i++;
 		}
-		
+
 		return $sharedProjects;
 	}
-	
+
 	public function getPublic() {
 		$query = $this->db->where(array('public' => '1'))
 			->order_by('name', 'asc')
@@ -56,10 +54,9 @@ class Project extends CI_Model {
 			$publicProjects[$i]['cls'] = 'folder';
 			$publicProjects[$i]['text'] = $project['name'];
 			$publicProjects[$i]['leaf'] = true;
-			$publicProjects[$i]['icon'] = "/ScattPort/assets/images/icons/document.png";
 			$i++;
 		}
-		
+
 		return $publicProjects;
 	}
 }
\ No newline at end of file