1 files changed, 3 insertions, 1 deletions
diff --git a/spec/helpers/tree_helper_spec.rb b/spec/helpers/tree_helper_spec.rb
index 1ca5b8eb954..c94844eebbc 100644
--- a/spec/helpers/tree_helper_spec.rb
+++ b/spec/helpers/tree_helper_spec.rb
@@ -21,12 +21,14 @@ RSpec.describe TreeHelper do
 
   describe '#vue_file_list_data' do
     it 'returns a list of attributes related to the project' do
+      helper.instance_variable_set(:@ref_type, 'heads')
       expect(helper.vue_file_list_data(project, sha)).to include(
         project_path: project.full_path,
         project_short_path: project.path,
         ref: sha,
         escaped_ref: sha,
-        full_name: project.name_with_namespace
+        full_name: project.name_with_namespace,
+        ref_type: 'heads'
       )
     end
   end