1 files changed, 7 insertions, 7 deletions
diff --git a/spec/features/refinery/blog/admin/categories_spec.rb b/spec/features/refinery/blog/admin/categories_spec.rb
index 907c6a0..b3e3f31 100644
--- a/spec/features/refinery/blog/admin/categories_spec.rb
+++ b/spec/features/refinery/blog/admin/categories_spec.rb
@@ -44,10 +44,10 @@ describe "Categories admin", type: :feature do
         expect(Refinery::Blog::Category.count).to eq(1)
       end
 
-      it "shows locale flag for category" do
+      it "shows locale for category" do
         click_link "Manage"
         within "#category_#{@c.id}" do
-          expect(page).to have_css("img[src='/assets/refinery/icons/flags/en.png']")
+          expect(page).to have_css(".locale_icon.en")
         end
       end
 
@@ -73,7 +73,7 @@ describe "Categories admin", type: :feature do
         visit refinery.blog_admin_posts_path
         click_link "Create new category"
         within "#switch_locale_picker" do
-          click_link "ru"
+          click_link "RU"
         end
         fill_in "Title", :with => ru_category_title
         click_button "Save"
@@ -85,17 +85,17 @@ describe "Categories admin", type: :feature do
         expect(Refinery::Blog::Category.count).to eq(1)
       end
 
-      it "shows locale flag for category" do
+      it "shows locale for category" do
         click_link "Manage"
         within "#category_#{@c.id}" do
-          expect(page).to have_css("img[src='/assets/refinery/icons/flags/ru.png']")
+          expect(page).to have_css(".locale_icon.ru")
         end
       end
 
-      it "does not show locale flag for primary locale" do
+      it "does not show locale for primary locale" do
         click_link "Manage"
         within "#category_#{@c.id}" do
-          expect(page).not_to have_css("img[src='/assets/refinery/icons/flags/en.png']")
+          expect(page).not_to have_css(".locale_icon.en")
         end
       end