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(timestamp, new Marker(null, pos))); + Keyframe markerKeyframe = new Keyframe(timestamp, new Marker(null, pos)); + markerKeyframes.add(markerKeyframe); + ReplayHandler.selectKeyframe(markerKeyframe); } }