From adfb6dee125fde6a7d06bd0b3030a289ba516697 Mon Sep 17 00:00:00 2001
From: Kevin Sawicki <kevin@github.com>
Date: Mon, 7 Jan 2013 18:21:03 -0800
Subject: [PATCH] :lipstick:

---
 src/packages/tree-view/src/directory-view.coffee | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/src/packages/tree-view/src/directory-view.coffee b/src/packages/tree-view/src/directory-view.coffee
index df177bba4..3614f79da 100644
--- a/src/packages/tree-view/src/directory-view.coffee
+++ b/src/packages/tree-view/src/directory-view.coffee
@@ -23,17 +23,15 @@ class DirectoryView extends View
     @disclosureArrow.on 'click', => @toggleExpansion()
 
     repo = @project.repo
+    iconClass = 'directory-icon'
     if repo?
       path = @directory.getPath()
-      @directoryName.addClass('ignored') if parent and repo.isPathIgnored(path)
-      if not parent and path is repo.getWorkingDirectory()
-        @directoryName.addClass('repository-icon')
-      else if repo.isSubmodule(path)
-        @directoryName.addClass('submodule-icon')
+      if parent
+        @directoryName.addClass('ignored') if repo.isPathIgnored(path)
+        iconClass = 'submodule-icon' if repo.isSubmodule(path)
       else
-        @directoryName.addClass('directory-icon')
-    else
-      @directoryName.addClass('directory-icon')
+        iconClass = 'repository-icon' if path is repo.getWorkingDirectory()
+    @directoryName.addClass(iconClass)
 
   getPath: ->
     @directory.path