diff --git a/spec/features/legislation_spec.rb b/spec/features/legislation_spec.rb
index d8df3351e..e3b2e7753 100644
--- a/spec/features/legislation_spec.rb
+++ b/spec/features/legislation_spec.rb
@@ -26,17 +26,13 @@ feature 'Legislation' do
       page.find(:css, ".annotator-controls a[href='#save']").click
 
       within ".annotate" do
-        expect(page).to have_css ".annotator-hl"
-      end
-
-      annotation = Annotation.last
-      within ".annotate" do
-        expect(page).to have_css ".annotator-hl[data-annotation-id='#{annotation.id}']"
+        expect(page).to have_css ".annotator-hl[data-annotation-id]"
       end
 
       visit legislation_path(legislation)
+
       within ".annotate" do
-        expect(page).to have_css ".annotator-hl[data-annotation-id='#{annotation.id}']"
+        expect(page).to have_css ".annotator-hl[data-annotation-id]"
       end
     end
 
@@ -59,4 +55,4 @@ feature 'Legislation' do
 
 end
 
-## Notes logged in and not logged in users
\ No newline at end of file
+## Notes logged in and not logged in users