1 files changed, 11 insertions, 1 deletions
diff --git a/app/workers/gitlab/bitbucket_import/advance_stage_worker.rb b/app/workers/gitlab/bitbucket_import/advance_stage_worker.rb
index 7f281352a1b..ed89f332652 100644
--- a/app/workers/gitlab/bitbucket_import/advance_stage_worker.rb
+++ b/app/workers/gitlab/bitbucket_import/advance_stage_worker.rb
@@ -20,13 +20,23 @@ module Gitlab
 
       # The known importer stages and their corresponding Sidekiq workers.
       STAGES = {
+        repository: Stage::ImportRepositoryWorker,
+        pull_requests: Stage::ImportPullRequestsWorker,
+        pull_requests_notes: Stage::ImportPullRequestsNotesWorker,
+        issues: Stage::ImportIssuesWorker,
+        issues_notes: Stage::ImportIssuesNotesWorker,
+        lfs_objects: Stage::ImportLfsObjectsWorker,
         finish: Stage::FinishImportWorker
       }.freeze
 
-      def find_import_state(project_id)
+      def find_import_state_jid(project_id)
         ProjectImportState.jid_by(project_id: project_id, status: :started)
       end
 
+      def find_import_state(id)
+        ProjectImportState.find(id)
+      end
+
       private
 
       def next_stage_worker(next_stage)