1 files changed, 9 insertions, 6 deletions
diff --git a/lib/gitlab/import_export/wiki_repo_saver.rb b/lib/gitlab/import_export/wiki_repo_saver.rb
index 93ae6f6b02a..4b1cf4915e4 100644
--- a/lib/gitlab/import_export/wiki_repo_saver.rb
+++ b/lib/gitlab/import_export/wiki_repo_saver.rb
@@ -3,18 +3,21 @@
 module Gitlab
   module ImportExport
     class WikiRepoSaver < RepoSaver
-      def save
-        wiki = ProjectWiki.new(project)
-        @repository = wiki.repository
+      extend ::Gitlab::Utils::Override
 
-        super
+      override :repository
+      def repository
+        @repository ||= exportable.wiki.repository
       end
 
       private
 
-      def bundle_full_path
-        File.join(shared.export_path, ImportExport.wiki_repo_bundle_filename)
+      override :bundle_filename
+      def bundle_filename
+        ::Gitlab::ImportExport.wiki_repo_bundle_filename
       end
     end
   end
 end
+
+Gitlab::ImportExport::WikiRepoSaver.prepend_if_ee('EE::Gitlab::ImportExport::WikiRepoSaver')