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