From a55173ffc70e105c0ad17685c4d680a2666c7120 Mon Sep 17 00:00:00 2001
From: CrushedPixel <crushedpixelmaps@gmail.com>
Date: Fri, 17 Jul 2015 02:30:23 +0200
Subject: [PATCH] Marker Keyframes are initially selected after creating them
 and get properly deselected when removing them

---
 .../crushedpixel/replaymod/replay/ReplayHandler.java   | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/main/java/eu/crushedpixel/replaymod/replay/ReplayHandler.java b/src/main/java/eu/crushedpixel/replaymod/replay/ReplayHandler.java
index bd687010..6e8058d6 100755
--- a/src/main/java/eu/crushedpixel/replaymod/replay/ReplayHandler.java
+++ b/src/main/java/eu/crushedpixel/replaymod/replay/ReplayHandler.java
@@ -228,11 +228,15 @@ public class ReplayHandler {
     }
 
     public static void toggleMarker() {
-        if(selectedKeyframe != null && selectedKeyframe.getValue() instanceof Marker) markerKeyframes.remove(selectedKeyframe);
-        else {
+        if(selectedKeyframe != null && selectedKeyframe.getValue() instanceof Marker) {
+            markerKeyframes.remove(selectedKeyframe);
+            ReplayHandler.selectKeyframe(null);
+        } else {
             AdvancedPosition pos = new AdvancedPosition(mc.getRenderViewEntity(), false);
             int timestamp = ReplayMod.replaySender.currentTimeStamp();
-            markerKeyframes.add(new Keyframe<Marker>(timestamp, new Marker(null, pos)));
+            Keyframe<Marker> markerKeyframe = new Keyframe<Marker>(timestamp, new Marker(null, pos));
+            markerKeyframes.add(markerKeyframe);
+            ReplayHandler.selectKeyframe(markerKeyframe);
         }
     }