From 16ea6a31fcaf69e2ed3702a1c48bc6e076481e01 Mon Sep 17 00:00:00 2001
From: achal7 <achal7@gmail.com>
Date: Fri, 13 Apr 2018 17:56:25 +0530
Subject: [PATCH] 1) changed UpdateLastState to call UpdateLastState(TState) so
 that its the only one that has last update control 2) Clear function clean up
 the states, however currentNode by definition is depends upon one of the
 state value, hence made that also null, since there is nothing now to link

---
 src/Avalonia.Controls/Utils/UndoRedoHelper.cs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Avalonia.Controls/Utils/UndoRedoHelper.cs b/src/Avalonia.Controls/Utils/UndoRedoHelper.cs
index c76555e554..17cf681f15 100644
--- a/src/Avalonia.Controls/Utils/UndoRedoHelper.cs
+++ b/src/Avalonia.Controls/Utils/UndoRedoHelper.cs
@@ -59,7 +59,7 @@ namespace Avalonia.Controls.Utils
 
         public void UpdateLastState()
         {
-            _states.Last.Value = _host.UndoRedoState;
+            UpdateLastState(_host.UndoRedoState);
         }
 
         public void DiscardRedo()
@@ -94,6 +94,7 @@ namespace Avalonia.Controls.Utils
         public void Clear()
         {
             _states.Clear();
+            _currentNode = null;
         }
 
         bool WeakTimer.IWeakTimerSubscriber.Tick()