diff --git a/spec/lib/gitlab/background_migration/populate_external_pipeline_source_spec.rb b/spec/lib/gitlab/background_migration/populate_external_pipeline_source_spec.rb
index d95077beb59..1a1193fa1fa 100644
--- a/spec/lib/gitlab/background_migration/populate_external_pipeline_source_spec.rb
+++ b/spec/lib/gitlab/background_migration/populate_external_pipeline_source_spec.rb
@@ -5,13 +5,14 @@ require 'spec_helper'
 describe Gitlab::BackgroundMigration::PopulateExternalPipelineSource, :migration, schema: 20180916011959 do
   let(:migration) { described_class.new }
 
-  let!(:internal_pipeline) { create(:ci_pipeline, source: Ci::Pipeline.sources[:web]) }
+  let!(:internal_pipeline) { create(:ci_pipeline, source: :web) }
+
   let!(:external_pipeline) do
-    build(:ci_pipeline, source: Ci::Pipeline.sources[:unknown])
+    build(:ci_pipeline, source: :unknown)
       .tap { |pipeline| pipeline.save(validate: false) }
   end
   let!(:second_external_pipeline) do
-    build(:ci_pipeline, source: Ci::Pipeline.sources[:unknown])
+    build(:ci_pipeline, source: :unknown)
       .tap { |pipeline| pipeline.save(validate: false) }
   end