diff --git a/spec/features/merge_requests/diff_notes_spec.rb b/spec/features/merge_requests/diff_notes_spec.rb
index 482c280c188e037cfd90259e3c015d7d2c53e008..a371d89f7a5a6c8ab9ac11c6a1ed922bfa3b5940 100644
--- a/spec/features/merge_requests/diff_notes_spec.rb
+++ b/spec/features/merge_requests/diff_notes_spec.rb
@@ -102,7 +102,7 @@ feature 'Diff notes', js: true, feature: true do
 
       context 'with an unfolded line' do
         before(:each) do
-          first('.js-unfold').click
+          find('.js-unfold', match: :first).click
           wait_for_ajax
         end