Can't Exit AnimationMode

I have a pretty standard animation preview script like is featured in the docs for AnimationMode doing stuff like

Code (CSharp):

1. AnimationMode.BeginSampling();
2. AnimationMode.SampleAnimationClip(animator.gameObject, _currentAnimation, _displayTime);
3. AnimationMode.EndSampling();
4. SceneView.RepaintAll();

and at the end I

Code (CSharp):

1. AnimationMode.StopAnimationMode();

Unfortunately I can't ever exit Animation Mode. The controller keeps a blue colored checkbox and my model doesn't revert to it's pre-animated state. I actually added a button

Code (CSharp):

1. if (AnimationMode.InAnimationMode() && GUILayout.Button("End Animation Mode")) {
2.       AnimationMode.StopAnimationMode();
3.  

and you can hammer it and nothing happens. How do I exit? This is 2018.3.0f2

Edit: Bah I restarted and immediately tried it and it was working properly. I think if it errors out even once (due to script or animation error) it'll be screwed until the next restart. So I need to restart whenever that happens.

 

Have you found a fix for this?