diff --git a/x-pack/test/security_solution_api_integration/test_suites/entity_analytics/risk_engine/trial_license_complete_tier/init_and_status_apis.ts b/x-pack/test/security_solution_api_integration/test_suites/entity_analytics/risk_engine/trial_license_complete_tier/init_and_status_apis.ts
index 19a9bb85326fa..3224caa24d5e2 100644
--- a/x-pack/test/security_solution_api_integration/test_suites/entity_analytics/risk_engine/trial_license_complete_tier/init_and_status_apis.ts
+++ b/x-pack/test/security_solution_api_integration/test_suites/entity_analytics/risk_engine/trial_license_complete_tier/init_and_status_apis.ts
@@ -19,6 +19,10 @@ import {
 } from '../../utils';
 import { FtrProviderContext } from '../../../../ftr_provider_context';
 
+const expectTaskIsNotRunning = (taskStatus?: string) => {
+  expect(['idle', 'claiming']).contain(taskStatus);
+};
+
 export default ({ getService }: FtrProviderContext) => {
   const es = getService('es');
   const supertest = getService('supertest');
@@ -355,7 +359,7 @@ export default ({ getService }: FtrProviderContext) => {
         expect(status2.body.legacy_risk_engine_status).to.be('NOT_INSTALLED');
 
         expect(status2.body.risk_engine_task_status?.runAt).to.be.a('string');
-        expect(status2.body.risk_engine_task_status?.status).to.be('idle');
+        expectTaskIsNotRunning(status2.body.risk_engine_task_status?.status);
         expect(status2.body.risk_engine_task_status?.startedAt).to.be(undefined);
 
         await riskEngineRoutes.disable();
@@ -373,7 +377,7 @@ export default ({ getService }: FtrProviderContext) => {
         expect(status4.body.legacy_risk_engine_status).to.be('NOT_INSTALLED');
 
         expect(status4.body.risk_engine_task_status?.runAt).to.be.a('string');
-        expect(status4.body.risk_engine_task_status?.status).to.be('idle');
+        expectTaskIsNotRunning(status4.body.risk_engine_task_status?.status);
         expect(status4.body.risk_engine_task_status?.startedAt).to.be(undefined);
       });
 
@@ -394,7 +398,7 @@ export default ({ getService }: FtrProviderContext) => {
         expect(status2.body.legacy_risk_engine_status).to.be('NOT_INSTALLED');
 
         expect(status2.body.risk_engine_task_status?.runAt).to.be.a('string');
-        expect(status2.body.risk_engine_task_status?.status).to.be('idle');
+        expectTaskIsNotRunning(status2.body.risk_engine_task_status?.status);
         expect(status2.body.risk_engine_task_status?.startedAt).to.be(undefined);
       });
     });