diff --git a/test_suites/live/src/StudioLiveDriverNode.cs b/test_suites/live/src/StudioLiveDriverNode.cs
index 52d0ad2..565edca 100644
--- a/test_suites/live/src/StudioLiveDriverNode.cs
+++ b/test_suites/live/src/StudioLiveDriverNode.cs
@@ -10,6 +10,7 @@ namespace Cthangover.Mods.ScenarioStudio.Tests.Live
     using Cthangover.Mods.Dialog;
     using Cthangover.Mods.Dialog.Action.Impls;
     using Cthangover.Mods.Dialog.Executable;
+    using Cthangover.Mods.Interface.View;
     using Godot;
 
     /// <summary>
@@ -598,7 +599,7 @@ namespace Cthangover.Mods.ScenarioStudio.Tests.Live
             var dir = ProjectSettings.GlobalizePath(_shotsDir);
             Directory.CreateDirectory(dir);
             var path = Path.Combine(dir, id + ".png");
-            GetViewport().GetTexture().GetImage().SavePng(path);
+            ViewportCapture.SavePng(GetViewport(), path);
             Log($"screenshot {path}");
         }
 
diff --git a/ui/icons/studio.png b/ui/icons/studio.png
index 705d68b..9f932cf 100644
Binary files a/ui/icons/studio.png and b/ui/icons/studio.png differ