From 86ce948fa03c4e6f5a6780bd9116146d6df25aa2 Mon Sep 17 00:00:00 2001
From: nathank33 <nathanbraunguitar@gmail.com>
Date: Tue, 9 Sep 2014 21:12:41 -0700
Subject: [PATCH] LavaFlow not cleaning while shift held.

Fixed a bug where LavaFlow was not being removed if the user kept the
shift button held down. Now it cleans up regardless of whether they are
still trying to bend a LavaFlow.
---
 .../projectkorra/ProjectKorra/earthbending/LavaFlow.java   | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/com/projectkorra/ProjectKorra/earthbending/LavaFlow.java b/src/com/projectkorra/ProjectKorra/earthbending/LavaFlow.java
index 2504ff25..7a8f5dda 100644
--- a/src/com/projectkorra/ProjectKorra/earthbending/LavaFlow.java
+++ b/src/com/projectkorra/ProjectKorra/earthbending/LavaFlow.java
@@ -143,7 +143,12 @@ public class LavaFlow
 			remove();
 			return;
 		}		
-
+		
+		double removeDelay = AvatarState.isAvatarState(player) ? AS_SHIFT_REMOVE_DELAY : SHIFT_REMOVE_DELAY;
+		if(System.currentTimeMillis() - time > removeDelay){
+			remove();
+			return;
+		}
 		
 		if(type == AbilityType.SHIFT)
 		{