Migrate pipeline stages only when not migrated already

1 files changed, 3 insertions, 3 deletions

diff --git a/db/post_migrate/20170526185842_migrate_pipeline_stages.rb b/db/post_migrate/20170526185842_migrate_pipeline_stages.rb
index 05e095f07cb..382791c4664 100644
--- a/db/post_migrate/20170526185842_migrate_pipeline_stages.rb
+++ b/db/post_migrate/20170526185842_migrate_pipeline_stages.rb
@@ -11,9 +11,9 @@ class MigratePipelineStages < ActiveRecord::Migration
     execute <<-SQL.strip_heredoc
       INSERT INTO ci_stages (project_id, pipeline_id, name)
         SELECT project_id, commit_id, stage FROM ci_builds
-          WHERE stage IS NOT NULL
-          GROUP BY project_id, commit_id, stage, stage_idx
-          ORDER BY stage_idx
+          WHERE stage IS NOT NULL AND stage_id IS NULL
+          GROUP BY project_id, commit_id, stage
+          ORDER BY MAX(stage_idx)
     SQL
   end