diff --git a/tools/aiplugin.test.ts b/tools/aiplugin.test.ts
index d6dd54f..ebdba72 100644
--- a/tools/aiplugin.test.ts
+++ b/tools/aiplugin.test.ts
@@ -1,5 +1,6 @@
 import { createAIPlugin } from './aiplugin';
 import { expect, it } from 'vitest';
+import { ToolInterface } from './tool';
 
 it('should create a new AIPlugin', async () => {
   const name = 'klarna';
@@ -7,7 +8,7 @@ it('should create a new AIPlugin', async () => {
 
   const { klarna } = await createAIPlugin({
     name, url
-  });
+  }) as ToolInterface
 
   const result = await klarna({});
 
diff --git a/tools/aiplugin.ts b/tools/aiplugin.ts
index dee82db..0d33998 100644
--- a/tools/aiplugin.ts
+++ b/tools/aiplugin.ts
@@ -18,7 +18,6 @@ async function createAIPlugin({
   name: string,
   url: string,
 }) {
-
   const paramsSchema = z.object({
     product: z.string().optional(),
   });