diff --git a/x-pack/plugins/actions/server/sub_action_framework/register.test.ts b/x-pack/plugins/actions/server/sub_action_framework/register.test.ts
index 28fc369fec6c3..a2460ef8bb3ee 100644
--- a/x-pack/plugins/actions/server/sub_action_framework/register.test.ts
+++ b/x-pack/plugins/actions/server/sub_action_framework/register.test.ts
@@ -36,8 +36,6 @@ describe('Registration', () => {
     getService: (serviceParams: ServiceParams<TestConfig, TestSecrets>) =>
       new TestSubActionConnector(serviceParams),
     renderParameterTemplates: mockRenderParameterTemplates,
-    postDeleteEventHandler: mockDeleteEventHandler,
-    preSaveEventHandler: mockSaveEventHandler,
   };
 
   const actionTypeRegistry = actionTypeRegistryMock.create();
@@ -51,7 +49,11 @@ describe('Registration', () => {
   it('registers the connector correctly', async () => {
     register<TestConfig, TestSecrets>({
       actionTypeRegistry,
-      connector,
+      connector: {
+        ...connector,
+        postDeleteEventHandler: mockDeleteEventHandler,
+        preSaveEventHandler: mockSaveEventHandler,
+      },
       configurationUtilities: mockedActionsConfig,
       logger,
     });
diff --git a/x-pack/test/plugin_api_integration/test_suites/task_manager/check_registered_task_types.ts b/x-pack/test/plugin_api_integration/test_suites/task_manager/check_registered_task_types.ts
index db0b9736ecdba..3f9dea917e634 100644
--- a/x-pack/test/plugin_api_integration/test_suites/task_manager/check_registered_task_types.ts
+++ b/x-pack/test/plugin_api_integration/test_suites/task_manager/check_registered_task_types.ts
@@ -59,9 +59,9 @@ export default function ({ getService }: FtrProviderContext) {
         'actions:.d3security',
         'actions:.email',
         'actions:.gemini',
-        'actions:.inference',
         'actions:.gen-ai',
         'actions:.index',
+        'actions:.inference',
         'actions:.jira',
         'actions:.observability-ai-assistant',
         'actions:.opsgenie',