2004-12-02 Lluis Sanchez Gual <lluis@novell.com>

[mono.git] / mcs / class / System.Web / System.Web.UI.WebControls / TreeNodeCollection.cs

diff --git a/mcs/class/System.Web/System.Web.UI.WebControls/TreeNodeCollection.cs b/mcs/class/System.Web/System.Web.UI.WebControls/TreeNodeCollection.cs
index 698790512109eb8626309152223e4481bf2215dc..79899ead9a63e43a9743f64be46a2477f068e965 100644 (file)
--- a/mcs/class/System.Web/System.Web.UI.WebControls/TreeNodeCollection.cs
+++ b/mcs/class/System.Web/System.Web.UI.WebControls/TreeNodeCollection.cs
@@ -78,6 +78,7 @@ namespace System.Web.UI.WebControls
                        child.SetParent (parent);
                        if (marked) {
                                child.TrackViewState ();
+                               child.SetDirty ();
                                dirty = true;
                        }
                }
@@ -92,6 +93,7 @@ namespace System.Web.UI.WebControls
                                ((TreeNode)items[n]).Index = n;
                        if (marked) {
                                child.TrackViewState ();
+                               child.SetDirty ();
                                dirty = true;
                        }
                }
@@ -163,7 +165,7 @@ namespace System.Web.UI.WebControls
                {
                        items.CopyTo (array, index);
                }
-               
+
                void IStateManager.LoadViewState (object state)
                {
                        if (state == null) return;
@@ -180,9 +182,8 @@ namespace System.Web.UI.WebControls
                                TreeNode node;
                                if (oi != -1) node = originalItems [oi];
                                else node = new TreeNode ();
+                               if (dirty) Add (node);
                                node.LoadViewState (pair.Second);
-                               if (dirty)
-                                       Add (node);
                        }
                }