diff --git a/app/views/projects/tree/show.html.haml b/app/views/projects/tree/show.html.haml
index 6b33493b7d..735ffae9e2 100644
--- a/app/views/projects/tree/show.html.haml
+++ b/app/views/projects/tree/show.html.haml
@@ -1,6 +1,6 @@
 %div.tree-ref-holder
   = render 'shared/ref_switcher', destination: 'tree', path: @path
 - if can? current_user, :download_code, @project
-  = render 'projects/repositories/download_archive', ref: @ref, btn_class: 'btn-group-small tree-ref-holder pull-right', split_button: true
+  = render 'projects/repositories/download_archive', ref: @ref, btn_class: 'btn-group-small pull-right', split_button: true
 %div#tree-holder.tree-holder
   = render "tree", tree: @tree
diff --git a/features/steps/project/browse_branches.rb b/features/steps/project/browse_branches.rb
index 30c8cef80c..5b5e8ba493 100644
--- a/features/steps/project/browse_branches.rb
+++ b/features/steps/project/browse_branches.rb
@@ -44,7 +44,7 @@ class ProjectBrowseBranches < Spinach::FeatureSteps
   end
 
   step 'I should see new branch created' do
-    within '.all-branches' do
+    within '.tree-ref-holder' do
       page.should have_content 'deploy_keys'
     end
   end