diff --git a/packages/server/src/automations/tests/steps/loop.spec.ts b/packages/server/src/automations/tests/steps/loop.spec.ts
index 9e8d82c1f0..402fbcda92 100644
--- a/packages/server/src/automations/tests/steps/loop.spec.ts
+++ b/packages/server/src/automations/tests/steps/loop.spec.ts
@@ -25,8 +25,8 @@ describe("Attempt to run a basic loop automation", () => {
     await config.api.row.save(table._id!, {})
   })
 
-  afterAll(() => {
-    automation.shutdown()
+  afterAll(async () => {
+    await automation.shutdown()
     config.end()
   })
 
diff --git a/packages/server/src/definitions/automations.ts b/packages/server/src/definitions/automations.ts
index bb472e01a3..a04b960ca5 100644
--- a/packages/server/src/definitions/automations.ts
+++ b/packages/server/src/definitions/automations.ts
@@ -1,7 +1,5 @@
 import {
-  AutomationStepResult,
   AutomationStepResultOutputs,
-  AutomationTriggerResult,
   AutomationTriggerResultOutputs,
   LoopStepType,
   UserBindings,