diff --git a/test/integration/default_theme_test.rb b/test/integration/default_theme_test.rb
index 08e067c1..8dc44003 100644
--- a/test/integration/default_theme_test.rb
+++ b/test/integration/default_theme_test.rb
@@ -86,7 +86,7 @@ def create_page_and_menu
       with_temp_content_directory do
         model = create_page_and_menu
         visit model.path
-        assert_has_css "ul.menu li[class='current']:contains('#{model.link_text}')"
+        assert_has_css "ul.menu li.current:contains('#{model.link_text}')"
       end
     end
 
@@ -100,7 +100,7 @@ def create_page_and_menu
       with_temp_content_directory do
         model = create_page_and_menu
         visit "/prefix/#{model.path}"
-        assert_has_css "ul.menu li[class='current']:contains('#{model.link_text}')"
+        assert_has_css "ul.menu li.current:contains('#{model.link_text}')"
       end
     end
   end