diff --git a/tests/extension.spec.ts b/tests/extension.spec.ts
index 74df147..6c654b4 100644
--- a/tests/extension.spec.ts
+++ b/tests/extension.spec.ts
@@ -126,6 +126,7 @@ test('blob preview with an .stl file', async ({
 }) => {
     const url =
         'https://github.com/KittyCAD/diff-samples/blob/fd9eec79f0464833686ea6b5b34ea07145e32734/models/box.stl'
-    const screenshot = await getBlobPreviewScreenshot(page, url)
+    const element = await getBlobPreviewElement(page, url)
+    const screenshot = await element.screenshot()
     expect(screenshot).toMatchSnapshot()
 })