diff --git a/test_suites/trade/src/TradeTestDriver.cs b/test_suites/trade/src/TradeTestDriver.cs
index e84bd74..4a2d730 100644
--- a/test_suites/trade/src/TradeTestDriver.cs
+++ b/test_suites/trade/src/TradeTestDriver.cs
@@ -10,6 +10,7 @@ using Cthangover.Core.UI.Inventory;
 using Cthangover.Core.Utils;
 using Cthangover.Mods.Items;
 using Cthangover.Mods.Rpg;
+using Cthangover.Mods.Interface.View;
 using Godot;
 
 namespace Cthangover.Mods.Trading.Tests
@@ -341,7 +342,7 @@ namespace Cthangover.Mods.Trading.Tests
             if (string.IsNullOrEmpty(shotsDir) || DisplayServer.GetName() == "headless")
                 return;
             DirAccess.MakeDirRecursiveAbsolute(shotsDir);
-            GetViewport().GetTexture().GetImage()?.SavePng(shotsDir.TrimEnd('/') + "/trade_" + name + ".png");
+            ViewportCapture.SavePng(GetViewport(), shotsDir.TrimEnd('/') + "/trade_" + name + ".png");
         }
 
         private void Finish()