diff --git a/src/services/screenshot.ts b/src/services/screenshot.ts
index 3eb85f6..84e2ee7 100644
--- a/src/services/screenshot.ts
+++ b/src/services/screenshot.ts
@@ -159,18 +159,7 @@ export async function takeScreenshot(opts: ScreenshotOptions): Promise<Screensho
     }
     if (quality !== undefined) screenshotOpts.quality = quality;
 
-    let result: Buffer;
-    if (opts.selector) {
-      // Take element screenshot
-      const element = await page.$(opts.selector);
-      if (!element) {
-        throw new Error("SELECTOR_NOT_FOUND");
-      }
-      result = await element.screenshot(screenshotOpts) as Buffer;
-    } else {
-      // Take page screenshot
-      result = await page.screenshot(screenshotOpts) as Buffer;
-    }
+    const result = await page.screenshot(screenshotOpts);
 
     const buffer = Buffer.from(result as unknown as ArrayBuffer);