1 files changed, 1 insertions, 3 deletions
diff --git a/lib/gitlab/background_migration/migrate_build_stage_id_reference.rb b/lib/gitlab/background_migration/migrate_build_stage_id_reference.rb
index c8669ca3272..d1d0a968588 100644
--- a/lib/gitlab/background_migration/migrate_build_stage_id_reference.rb
+++ b/lib/gitlab/background_migration/migrate_build_stage_id_reference.rb
@@ -2,15 +2,13 @@ module Gitlab
   module BackgroundMigration
     class MigrateBuildStageIdReference
       def perform(id)
-        raise ArgumentError unless id.present?
-
         sql = <<-SQL.strip_heredoc
           UPDATE "ci_builds"
             SET "stage_id" =
               (SELECT id FROM ci_stages
                 WHERE ci_stages.pipeline_id = ci_builds.commit_id
                 AND ci_stages.name = ci_builds.stage)
-          WHERE "ci_builds"."id" = #{id}
+          WHERE "ci_builds"."id" = #{id.to_i}
             AND "ci_builds"."stage_id" IS NULL
         SQL