diff --git a/package.json b/package.json
index d76d9cd..18c08ba 100644
--- a/package.json
+++ b/package.json
@@ -3,7 +3,7 @@
   "description": "SolidJS tRPC",
   "author": "OrJDev",
   "license": "MIT",
-  "version": "0.0.8-rc.1",
+  "version": "0.0.9-rc.1",
   "publishConfig": {
     "access": "public",
     "tag": "next"
diff --git a/src/shared/proxy/decorationProxy.ts b/src/shared/proxy/decorationProxy.ts
index 9829e97..ee7e776 100644
--- a/src/shared/proxy/decorationProxy.ts
+++ b/src/shared/proxy/decorationProxy.ts
@@ -26,7 +26,8 @@ export function createSolidProxyDecoration<
       return (hooks as any)[lastArg](path, ...args);
     }
     return (hooks as any)[lastArg](
-      () => getQueryKey(path, (args[0] as any)()),
+      () =>
+        getQueryKey(path, typeof args[0] === "function" ? args[0]() : args[0]),
       args[1]
     );
   });