diff --git a/packages/render-fidelity-tools/src/artifact-creator.ts b/packages/render-fidelity-tools/src/artifact-creator.ts
index b11bebf510..2685aae39d 100644
--- a/packages/render-fidelity-tools/src/artifact-creator.ts
+++ b/packages/render-fidelity-tools/src/artifact-creator.ts
@@ -382,7 +382,9 @@ export class ArtifactCreator {
     }
 
     const screenshot =
-        await page.screenshot({path: outputPath as `${string}.png`, omitBackground: true});
+        outputPath && (outputPath.endsWith('.png') || outputPath.endsWith('.jpeg') || outputPath.endsWith('.webp'))
+            ? await page.screenshot({path: outputPath as `${string}.png` | `${string}.jpeg` | `${string}.webp`, omitBackground: true})
+            : await page.screenshot({omitBackground: true});
 
     page.close();