1 files changed, 3 insertions, 1 deletions
diff --git a/spec/requests/api/graphql/ci/runners_spec.rb b/spec/requests/api/graphql/ci/runners_spec.rb
index a5b8115286e..749f6839cb5 100644
--- a/spec/requests/api/graphql/ci/runners_spec.rb
+++ b/spec/requests/api/graphql/ci/runners_spec.rb
@@ -37,7 +37,9 @@ RSpec.describe 'Query.runners' do
     end
 
     before do
-      allow(Gitlab::Ci::RunnerUpgradeCheck.instance).to receive(:check_runner_upgrade_status)
+      allow_next_instance_of(::Gitlab::Ci::RunnerUpgradeCheck) do |instance|
+        allow(instance).to receive(:check_runner_upgrade_suggestion)
+      end
 
       post_graphql(query, current_user: current_user)
     end