diff --git a/js/jquery.treegrid.js b/js/jquery.treegrid.js
index f605f13..ad0e0b5 100644
--- a/js/jquery.treegrid.js
+++ b/js/jquery.treegrid.js
@@ -144,16 +144,21 @@
          * @returns {Node}
          */
         initState: function() {
-            var $this = $(this);
+            var $this = $(this),
+                initialState = '';
+
             if ($this.treegrid('getSetting', 'saveState') && !$this.treegrid('isFirstInit')) {
                 $this.treegrid('restoreState');
             } else {
-                if ($this.treegrid('getSetting', 'initialState') === "expanded") {
+                initialState = $this.treegrid('getSetting', 'initialState');
+
+                if (initialState === "expanded") {
                     $this.treegrid('expand');
-                } else {
+                } else if (initialState === "collapse") {
                     $this.treegrid('collapse');
                 }
             }
+            
             return $this;
         },
         /**