1 files changed, 5 insertions, 6 deletions
diff --git a/spec/lib/gitlab/gl_repository_spec.rb b/spec/lib/gitlab/gl_repository_spec.rb
index 7be01507a82..b03edcb31a6 100644
--- a/spec/lib/gitlab/gl_repository_spec.rb
+++ b/spec/lib/gitlab/gl_repository_spec.rb
@@ -4,9 +4,8 @@ require 'spec_helper'
 
 RSpec.describe ::Gitlab::GlRepository do
   describe '.parse' do
-    let_it_be(:project) { create(:project, :repository) }
+    let_it_be(:project) { create(:project_with_design, :repository) }
     let_it_be(:snippet) { create(:personal_snippet) }
-    let(:design_repository_container) { project.design_repository.container }
 
     it 'parses a project gl_repository' do
       expect(described_class.parse("project-#{project.id}")).to eq([project, project, Gitlab::GlRepository::PROJECT])
@@ -21,11 +20,11 @@ RSpec.describe ::Gitlab::GlRepository do
     end
 
     it 'parses a design gl_repository' do
-      expect(described_class.parse("design-#{design_repository_container.id}")).to eq(
+      expect(described_class.parse("design-#{project.design_management_repository.id}")).to eq(
         [
-          design_repository_container,
-          project,
-          Gitlab::GlRepository::DESIGN
+          project.design_management_repository, # container
+          project,                              # project for container
+          Gitlab::GlRepository::DESIGN          # repo type
         ]
       )
     end