1 files changed, 7 insertions, 5 deletions
diff --git a/lib/gitlab/import_export/design_repo_saver.rb b/lib/gitlab/import_export/design_repo_saver.rb
index db9ebee6a13..b400aedc205 100644
--- a/lib/gitlab/import_export/design_repo_saver.rb
+++ b/lib/gitlab/import_export/design_repo_saver.rb
@@ -3,16 +3,18 @@
 module Gitlab
   module ImportExport
     class DesignRepoSaver < RepoSaver
-      def save
-        @repository = project.design_repository
+      extend ::Gitlab::Utils::Override
 
-        super
+      override :repository
+      def repository
+        @repository ||= exportable.design_repository
       end
 
       private
 
-      def bundle_full_path
-        File.join(shared.export_path, ::Gitlab::ImportExport.design_repo_bundle_filename)
+      override :bundle_filename
+      def bundle_filename
+        ::Gitlab::ImportExport.design_repo_bundle_filename
       end
     end
   end