Fix specs

2 files changed, 2 insertions, 1 deletions

diff --git a/lib/ci/api/builds.rb b/lib/ci/api/builds.rb
index 8887bba1dbe..7d566faf7ed 100644
--- a/lib/ci/api/builds.rb
+++ b/lib/ci/api/builds.rb
@@ -18,6 +18,7 @@ module Ci
 
           if current_runner.is_runner_queue_value_latest?(params[:last_update])
             header 'X-GitLab-Last-Update', params[:last_update]
+            Gitlab::Metrics.add_event(:build_not_found_cached)
             return build_not_found!
           end
 
diff --git a/spec/services/ci/register_build_service_spec.rb b/spec/services/ci/register_build_service_spec.rb
index 27d7853bbdd..635ddf70c8d 100644
--- a/spec/services/ci/register_build_service_spec.rb
+++ b/spec/services/ci/register_build_service_spec.rb
@@ -171,7 +171,7 @@ module Ci
       end
 
       def execute(runner)
-        described_class.new(runner).execute
+        described_class.new(runner).execute.first
       end
     end
   end