Index: chrome/browser/dom_distiller/tab_utils_browsertest.cc |
diff --git a/chrome/browser/dom_distiller/tab_utils_browsertest.cc b/chrome/browser/dom_distiller/tab_utils_browsertest.cc |
index 37bd4f293b607ccbbc0b0c22f97713961941f7a6..c3c0c910f31f95dd0ef3bdb24c20219964a9abb7 100644 |
--- a/chrome/browser/dom_distiller/tab_utils_browsertest.cc |
+++ b/chrome/browser/dom_distiller/tab_utils_browsertest.cc |
@@ -45,12 +45,10 @@ class WebContentsMainFrameHelper : public content::WebContentsObserver { |
content::WebContentsObserver::Observe(web_contents); |
} |
- virtual void DidFinishLoad( |
- int64 frame_id, |
- const GURL& validated_url, |
- bool is_main_frame, |
- content::RenderViewHost* render_view_host) OVERRIDE { |
- if (is_main_frame && validated_url.scheme() == chrome::kDomDistillerScheme) |
+ virtual void DidFinishLoad(content::RenderFrameHost* render_frame_host, |
+ const GURL& validated_url) OVERRIDE { |
+ if (!render_frame_host->GetParent() && |
+ validated_url.scheme() == chrome::kDomDistillerScheme) |
callback_.Run(); |
} |