1 files changed, 4 insertions, 1 deletions
diff --git a/app/models/ci/build_need.rb b/app/models/ci/build_need.rb
index d4cbbfac4ab..3fa17d6d286 100644
--- a/app/models/ci/build_need.rb
+++ b/app/models/ci/build_need.rb
@@ -2,15 +2,18 @@
 
 module Ci
   class BuildNeed < Ci::ApplicationRecord
+    include Ci::Partitionable
     include BulkInsertSafe
 
     belongs_to :build, class_name: "Ci::Processable", foreign_key: :build_id, inverse_of: :needs
 
+    partitionable scope: :build
+
     validates :build, presence: true
     validates :name, presence: true, length: { maximum: 128 }
     validates :optional, inclusion: { in: [true, false] }
 
-    scope :scoped_build, -> { where('ci_builds.id=ci_build_needs.build_id') }
+    scope :scoped_build, -> { where("#{Ci::Build.quoted_table_name}.id = #{quoted_table_name}.build_id") }
     scope :artifacts, -> { where(artifacts: true) }
   end
 end