diff --git a/lumaai/functions/generation.ts b/lumaai/functions/generation.ts
index 5c68519..c9129a4 100644
--- a/lumaai/functions/generation.ts
+++ b/lumaai/functions/generation.ts
@@ -1,6 +1,6 @@
 import { log } from "@restackio/restack-sdk-ts/function";
 import LumaAI from "lumaai";
-import { lumaaiClient } from "lumaai/utils/client";
+import { lumaaiClient } from "../utils/client";
 
 export async function lumaaiGeneration({
   prompt,
diff --git a/lumaai/package.json b/lumaai/package.json
index 39bd6df..8b6c502 100644
--- a/lumaai/package.json
+++ b/lumaai/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@restackio/integrations-lumaai",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "main": "dist/service.js",
   "types": "dist/service.d.ts",
   "scripts": {