Index: tools/telemetry/telemetry/core/backends/webdriver/webdriver_tab_backend.py |
diff --git a/tools/telemetry/telemetry/core/backends/webdriver/webdriver_tab_backend.py b/tools/telemetry/telemetry/core/backends/webdriver/webdriver_tab_backend.py |
index d4dd75c0d5ee72b0bc1c19ad90e6f4db6e7353da..991e9b55ac9fe39d453708ede38b5ad3a95e0218 100644 |
--- a/tools/telemetry/telemetry/core/backends/webdriver/webdriver_tab_backend.py |
+++ b/tools/telemetry/telemetry/core/backends/webdriver/webdriver_tab_backend.py |
@@ -4,7 +4,7 @@ |
import logging |
-from telemetry.core.backends import png_bitmap |
+from telemetry.core import bitmap |
class WebDriverTabBackend(object): |
def __init__(self, browser_backend, window_handle): |
@@ -53,7 +53,7 @@ class WebDriverTabBackend(object): |
self._browser_backend.driver.switch_to_window(self._window_handle) |
snap = self._browser_backend.driver.get_screenshot_as_base64() |
if snap: |
- return png_bitmap.PngBitmap(snap) |
+ return bitmap.Bitmap(snap) |
return None |
@property |