diff --git a/spec/models/build_spec.rb b/spec/models/build_spec.rb
index 47ba4931460253f6acd2321893b9711f7f72713b..c7c247189f5be0a84613cfbaabf73e42434caa4c 100644
--- a/spec/models/build_spec.rb
+++ b/spec/models/build_spec.rb
@@ -693,9 +693,9 @@ describe Ci::Build, models: true do
 
   describe 'Project#builds_for' do
     it 'returns builds from ref and build name' do
-      latest_build = project.builds_for(build.name, 'HEAD').latest.first
+      build_ids = project.builds_for(build.name, 'HEAD').map(&:id)
 
-      expect(latest_build.id).to eq(build.id)
+      expect(build_ids).to eq([build.id])
     end
   end
 end
diff --git a/spec/models/project_spec.rb b/spec/models/project_spec.rb
index 06d99240708810e79144b9ada565a466df446869..143fd5167a4ff8890276adaf9a7c7a5303cf3f74 100644
--- a/spec/models/project_spec.rb
+++ b/spec/models/project_spec.rb
@@ -364,20 +364,6 @@ describe Project, models: true do
     end
   end
 
-  describe '#builds_for' do
-    let(:project) { create(:project) }
-    let(:pipeline) do
-      create(:ci_pipeline, project: project, sha: project.commit.sha)
-    end
-    let(:build) { create(:ci_build, pipeline: pipeline) }
-
-    it 'returns builds for a particular ref' do
-      build_ids = project.builds_for(build.name, build.sha).map(&:id)
-
-      expect(build_ids).to eq([build.id])
-    end
-  end
-
   describe :default_issues_tracker? do
     let(:project) { create(:project) }
     let(:ext_project) { create(:redmine_project) }