diff --git a/spec/models/build_spec.rb b/spec/models/build_spec.rb
index b2dcbd8da2e71af6266d9aa062229b40e474951f..a57f0b6886c40a9f10a750d4ae66e818458b2968 100644
--- a/spec/models/build_spec.rb
+++ b/spec/models/build_spec.rb
@@ -698,9 +698,9 @@ describe Ci::Build, models: true do
     end
 
     it 'returns builds from ref' do
-      build_ids = project.latest_success_builds_for('HEAD').map(&:id)
+      builds = project.latest_success_builds_for('HEAD')
 
-      expect(build_ids).to eq([build.id])
+      expect(builds).to contain_exactly(build)
     end
 
     it 'returns empty relation if the build cannot be found' do