1 files changed, 3 insertions, 2 deletions
diff --git a/spec/lib/gitlab/graphql/representation/submodule_tree_entry_spec.rb b/spec/lib/gitlab/graphql/representation/submodule_tree_entry_spec.rb
index a3ff75f17e6..28056a6085d 100644
--- a/spec/lib/gitlab/graphql/representation/submodule_tree_entry_spec.rb
+++ b/spec/lib/gitlab/graphql/representation/submodule_tree_entry_spec.rb
@@ -7,8 +7,10 @@ describe Gitlab::Graphql::Representation::SubmoduleTreeEntry do
   let(:repository) { project.repository }
 
   describe '.decorate' do
+    let(:submodules) { repository.tree.submodules }
+
     it 'returns array of SubmoduleTreeEntry' do
-      entries = described_class.decorate(repository.tree)
+      entries = described_class.decorate(submodules, repository.tree)
 
       expect(entries.first).to be_a(described_class)
 
@@ -26,4 +28,3 @@ describe Gitlab::Graphql::Representation::SubmoduleTreeEntry do
     end
   end
 end
-