diff --git a/test_suites/display_settings/src/DisplaySettingsTestDriverNode.cs b/test_suites/display_settings/src/DisplaySettingsTestDriverNode.cs
index 0c76530..26a2542 100644
--- a/test_suites/display_settings/src/DisplaySettingsTestDriverNode.cs
+++ b/test_suites/display_settings/src/DisplaySettingsTestDriverNode.cs
@@ -6,6 +6,7 @@ using Cthangover.Core.Bootstrap;
 using Cthangover.Core.Mods;
 using Cthangover.Core.Settings;
 using Cthangover.Core.Utils;
+using Cthangover.Mods.Interface.View;
 using Godot;
 
 namespace Mods.DisplaySettingsTest
@@ -203,7 +204,7 @@ namespace Mods.DisplaySettingsTest
             var dir = ProjectSettings.GlobalizePath(shotsDir);
             Directory.CreateDirectory(dir);
             var path = Path.Combine(dir, name + ".png");
-            GetViewport().GetTexture().GetImage().SavePng(path);
+            ViewportCapture.SavePng(GetViewport(), path);
             GameLogger.Log("TEST", $"DisplaySettingsTest: screenshot {path}");
         }
 
diff --git a/test_suites/menu_flow/src/MenuFlowTestDriverNode.cs b/test_suites/menu_flow/src/MenuFlowTestDriverNode.cs
index a7810ee..bd014ef 100644
--- a/test_suites/menu_flow/src/MenuFlowTestDriverNode.cs
+++ b/test_suites/menu_flow/src/MenuFlowTestDriverNode.cs
@@ -9,6 +9,7 @@ using Cthangover.Core.Games;
 using Cthangover.Core.Mods;
 using Cthangover.Core.Settings;
 using Cthangover.Core.Utils;
+using Cthangover.Mods.Interface.View;
 using Godot;
 
 namespace Mods.MenuFlowTest
@@ -191,7 +192,7 @@ namespace Mods.MenuFlowTest
                 return;
             dir = ProjectSettings.GlobalizePath(dir);
             Directory.CreateDirectory(dir);
-            GetViewport().GetTexture().GetImage().SavePng(Path.Combine(dir, name + ".png"));
+            ViewportCapture.SavePng(GetViewport(), Path.Combine(dir, name + ".png"));
         }
 
         private Button DeleteButton()