diff --git a/AssetStudioGUI/Components/TreeViewExtensions.cs b/AssetStudioGUI/Components/TreeViewExtensions.cs
index d06bfa7..e166280 100644
--- a/AssetStudioGUI/Components/TreeViewExtensions.cs
+++ b/AssetStudioGUI/Components/TreeViewExtensions.cs
@@ -40,7 +40,7 @@ namespace AssetStudioGUI
                 hItem = node.Handle,
                 mask = TVIF_STATE,
                 stateMask = TVIS_STATEIMAGEMASK,
-                state = 0
+                state = TVIS_STATEIMAGEMASK
             };
             SendMessage(node.TreeView.Handle, TVM_SETITEM, IntPtr.Zero, ref tvi);
         }