Even more tests fixed :)

Signed-off-by: Dmitriy Zaporozhets <dmitriy.zaporozhets@gmail.com>

features/steps/shared/diff_note.rb

@@ -138,7 +138,7 @@ module SharedDiffNote
 
   Then 'I should see the diff comment edit button' do
     within(".diff-file") do
-      page.should have_css(".js-note-edit-button", visible: true)
+      page.should have_css(".js-note-write-button", visible: true)
     end
   end
 

features/steps/shared/note.rb

@@ -81,7 +81,7 @@ module SharedNote
 
   Then 'I should see the comment edit button' do
     within(".js-main-target-form") do
-      page.should have_css(".js-note-edit-button", visible: true)
+      page.should have_css(".js-note-write-button", visible: true)
     end
   end