1 files changed, 8 insertions, 6 deletions
diff --git a/features/steps/project/merge_requests.rb b/features/steps/project/merge_requests.rb
index da848afd48e..77602646d06 100644
--- a/features/steps/project/merge_requests.rb
+++ b/features/steps/project/merge_requests.rb
@@ -401,15 +401,17 @@ class Spinach::Features::ProjectMergeRequests < Spinach::FeatureSteps
     end
   end
 
-  step 'I click link "Hide inline discussion" of the third file' do
-    page.within '.files [id^=diff]:nth-child(3)' do
-      find('.js-toggle-diff-comments').trigger('click')
+  step 'I click link "Hide inline discussion" button' do
+    page.within '#diffs' do
+      find('.inline-parallel-buttons [data-toggle=dropdown]').trigger('click')
+      find('.js-diff-comments-button').trigger('click')
     end
   end
 
-  step 'I click link "Show inline discussion" of the third file' do
-    page.within '.files [id^=diff]:nth-child(3)' do
-      find('.js-toggle-diff-comments').trigger('click')
+  step 'I click link "Show inline discussion" button' do
+    page.within '#diffs' do
+      find('.inline-parallel-buttons [data-toggle=dropdown]').trigger('click')
+      find('.js-diff-comments-button').trigger('click')
     end
   end