From fa7256c58a8e98a9b5b774367f1cdc9dde26040b Mon Sep 17 00:00:00 2001
From: Xiaro <62033805+Xiaro@users.noreply.github.com>
Date: Fri, 20 Nov 2020 13:25:20 -0500
Subject: [PATCH] [fix] Fixed dying in Freecam not resetting the view entity
 correctly.

closes #1596
---
 .../zeroeightsix/kami/module/modules/player/Freecam.kt | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/main/java/me/zeroeightsix/kami/module/modules/player/Freecam.kt b/src/main/java/me/zeroeightsix/kami/module/modules/player/Freecam.kt
index 144cc4fc1..6c59af377 100644
--- a/src/main/java/me/zeroeightsix/kami/module/modules/player/Freecam.kt
+++ b/src/main/java/me/zeroeightsix/kami/module/modules/player/Freecam.kt
@@ -18,6 +18,7 @@ import net.minecraft.entity.player.EntityPlayer
 import net.minecraft.network.play.client.CPacketUseEntity
 import net.minecraft.util.math.Vec3d
 import net.minecraftforge.fml.common.gameevent.InputEvent
+import net.minecraftforge.fml.common.gameevent.TickEvent
 import org.lwjgl.input.Keyboard
 import kotlin.math.abs
 import kotlin.math.cos
@@ -69,9 +70,16 @@ object Freecam : Module() {
         }
 
         listener<SafeTickEvent> {
+            if (it.phase != TickEvent.Phase.END) return@listener
+
+            if (mc.player.isDead || mc.player.health <= 0.0f) {
+                if (cameraGuy != null) onDisable() // Reset the view entity, but not disable it
+                return@listener
+            }
+
             if (cameraGuy == null && mc.player.ticksExisted > 20) spawnCameraGuy()
 
-            if (!BaritoneUtils.isPathing) {
+            if (!BaritoneUtils.isPathing && !BaritoneUtils.isActive) {
                 if (autoRotate.value) updatePlayerRotation()
 
                 if (arrowKeyMove.value) updatePlayerMovement()