BE review changes

2 files changed, 2 insertions, 3 deletions

diff --git a/app/models/blob.rb b/app/models/blob.rb
index 91cf171a688..82333b6f369 100644
--- a/app/models/blob.rb
+++ b/app/models/blob.rb
@@ -63,7 +63,7 @@ class Blob < SimpleDelegator
   end
 
   def balsamiq?
-    binary? && extname.downcase.delete('.') == 'bmpr'
+    binary? && extension == 'bmpr'
   end
 
   def stl?
diff --git a/spec/features/projects/blobs/balsamiq_preview_spec.rb b/spec/features/projects/blobs/balsamiq_preview_spec.rb
index 551d2045744..59475525b81 100644
--- a/spec/features/projects/blobs/balsamiq_preview_spec.rb
+++ b/spec/features/projects/blobs/balsamiq_preview_spec.rb
@@ -7,7 +7,6 @@ feature 'Balsamiq preview', :feature, :js do
   let(:project) { create(:project) }
   let(:branch) { 'add-balsamiq-file' }
   let(:path) { 'files/images/balsamiq.bmpr' }
-  let(:file_content) { find('.file-content') }
 
   before do
     project.add_master(user)
@@ -16,7 +15,7 @@ feature 'Balsamiq preview', :feature, :js do
   end
 
   it 'should show a loading icon' do
-    expect(file_content).to have_selector('.loading')
+    expect(find('.file-content')).to have_selector('.loading')
   end
 
   it 'should show a viewer container' do