diff --git a/Assets/MagnetismHeroes/Tests/PlayMode/VisualUpgradePlayTests.cs b/Assets/MagnetismHeroes/Tests/PlayMode/VisualUpgradePlayTests.cs
index bcdda89..d42f063 100644
--- a/Assets/MagnetismHeroes/Tests/PlayMode/VisualUpgradePlayTests.cs
+++ b/Assets/MagnetismHeroes/Tests/PlayMode/VisualUpgradePlayTests.cs
@@ -285,7 +285,8 @@ namespace MagnetismHeroes.Tests
                 camera.targetTexture = target; camera.aspect = (float)width / height;
                 canvas.renderMode = RenderMode.ScreenSpaceCamera; canvas.worldCamera = camera; canvas.planeDistance = .5f;
                 foreach (var ball in _app.Presenter.Balls.Values) ball.SendMessage("Update");
-                _app.Presenter.SendMessage("LateUpdate"); Canvas.ForceUpdateCanvases();
+                if (_app.Presenter.gameObject.activeInHierarchy) _app.Presenter.SendMessage("LateUpdate");
+                Canvas.ForceUpdateCanvases();
                 var timer = System.Diagnostics.Stopwatch.StartNew(); camera.Render(); timer.Stop();
                 RenderTexture.active = target; image.ReadPixels(new Rect(0, 0, width, height), 0, 0); image.Apply();
                 File.WriteAllBytes(Path.Combine(_evidence, name + ".png"), image.EncodeToPNG());
@@ -312,7 +313,8 @@ namespace MagnetismHeroes.Tests
                 camera.targetTexture = oldTarget; camera.aspect = oldAspect; RenderTexture.active = oldActive;
                 canvas.renderMode = oldMode; canvas.worldCamera = oldCamera; canvas.planeDistance = oldPlane;
                 Object.DestroyImmediate(image); target.Release(); Object.DestroyImmediate(target);
-                _app.Presenter.SendMessage("LateUpdate"); Canvas.ForceUpdateCanvases();
+                if (_app.Presenter.gameObject.activeInHierarchy) _app.Presenter.SendMessage("LateUpdate");
+                Canvas.ForceUpdateCanvases();
             }
         }
     }