1 files changed, 6 insertions, 4 deletions
diff --git a/Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs b/Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs
index 3f5d20d4..ab755b64 100644
--- a/Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs
+++ b/Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs
@@ -6,17 +6,19 @@ using UnityEngine;
 public class WaitForActionReachEnd : IEnumerator
 {
 	UnitAnimation m_UnitAnimation;
-
-	public WaitForActionReachEnd(UnitAnimation unitAnim)
+    UnitAnimation.ELayer m_Layer;
+
+    public WaitForActionReachEnd(UnitAnimation unitAnim, UnitAnimation.ELayer layer = UnitAnimation.ELayer.Basic)
 	{
 		m_UnitAnimation = unitAnim;
-	}
+        m_Layer = layer;
+    }
 
 	public object Current => null;
 
 	public bool MoveNext()
 	{
-        var stateInfo = m_UnitAnimation.stateInfo;
+        var stateInfo = m_UnitAnimation.layers[(int)m_Layer].stateInfo;
         float normalTime = stateInfo.normalizedTime;
         LogHelper.Log(stateInfo.loop.ToString());
 		return normalTime < 1f;