diff --git a/tests/browser/features/references.feature b/tests/browser/features/references.feature
index 2273ddc..93c0ec4 100644
--- a/tests/browser/features/references.feature
+++ b/tests/browser/features/references.feature
@@ -1,15 +1,10 @@
-@chrome @en.m.wikipedia.beta.wmflabs.org @firefox @test2.m.wikipedia.org @vagrant @extension-cite
+@chrome @en.m.wikipedia.beta.wmflabs.org @firefox @test2.m.wikipedia.org @vagrant @integration @extension-cite
 Feature: Reference popup drawer
 
   Background:
     Given I am using the mobile site
 
-  Scenario: Opening the reference drawer
-    Given I go to a page that has references
-    When I click on a reference
-    Then I should see the reference drawer
-
-  Scenario: Closing the reference drawer
+  Scenario: Opening and closing the reference drawer
     Given I go to a page that has references
     When I click on a reference
       And I click on the page
diff --git a/tests/browser/features/support/pages/article_page.rb b/tests/browser/features/support/pages/article_page.rb
index 44eb848..c3b6c74 100644
--- a/tests/browser/features/support/pages/article_page.rb
+++ b/tests/browser/features/support/pages/article_page.rb
@@ -148,7 +148,7 @@ class ArticlePage
   div(:toast, css: '.mw-notification')
 
   # loader
-  div(:content_wrapper, id: 'content')
+  element(:content_wrapper, 'main')
   div(:content, id: 'bodyContent')
 
   # secondary menu