diff --git a/visual_regression_tracker/playwright.py b/visual_regression_tracker/playwright.py
index dfb6da6..21cc7e5 100644
--- a/visual_regression_tracker/playwright.py
+++ b/visual_regression_tracker/playwright.py
@@ -66,7 +66,7 @@ def trackPage(
         screenshot = page.screenshot(**screenshotOptions)
         imageBase64 = base64.b64encode(screenshot).decode('ascii')
 
-        self.track(TestRun(
+        return self.track(TestRun(
             name,
             imageBase64,
             options.agent.os if options and options.agent else None,
@@ -89,7 +89,7 @@ async def trackPageAsync(
         screenshot = await page.screenshot(**screenshotOptions)
         imageBase64 = base64.b64encode(screenshot).decode('ascii')
 
-        self.track(TestRun(
+        return self.track(TestRun(
             name,
             imageBase64,
             options.agent.os if options and options.agent else None,
@@ -110,7 +110,7 @@ def trackElementHandle(
         screenshot = elementHandle.screenshot(**screenshotOptions)
         imageBase64 = base64.b64encode(screenshot).decode('ascii')
 
-        self.track(TestRun(
+        return self.track(TestRun(
             name,
             imageBase64,
             options.agent.os if options and options.agent else None,
@@ -131,7 +131,7 @@ async def trackElementHandleAsync(
         screenshot = await elementHandle.screenshot(**screenshotOptions)
         imageBase64 = base64.b64encode(screenshot).decode('ascii')
 
-        self.track(TestRun(
+        return self.track(TestRun(
             name,
             imageBase64,
             options.agent.os if options and options.agent else None,