diff --git a/index.test-d.ts b/index.test-d.ts
index d0ac827..bef53c6 100644
--- a/index.test-d.ts
+++ b/index.test-d.ts
@@ -11,12 +11,15 @@ interface KirbySite {
 }
 
 // Query
+expectAssignable<KirbyQuery>("site");
 expectAssignable<KirbyQuery>('kirby.page("about")');
 expectAssignable<KirbyQuery>('collection("notes")');
+expectAssignable<KirbyQuery<"custom">>("custom");
+expectAssignable<KirbyQuery<"custom">>("custom.cover");
 expectNotAssignable<KirbyQuery>("kirby(");
 expectNotAssignable<KirbyQuery>('kirby("about');
 
-// Query Request
+// KQL Query Request
 expectAssignable<KirbyQueryRequest>({
   query: "site",
   select: {
@@ -32,7 +35,7 @@ expectAssignable<KirbyQueryRequest>({
   },
 });
 
-// Query Response
+// KQL Query Response
 expectAssignable<KirbyQueryResponse<KirbySite>>({
   code: 200,
   status: "OK",
diff --git a/src/query.d.ts b/src/query.d.ts
index 416145c..777415e 100644
--- a/src/query.d.ts
+++ b/src/query.d.ts
@@ -1,13 +1,14 @@
 // https://github.com/getkirby/kql/blob/66abd20093e5656b0f7e6f51ee04f630ab38f2a3/src/Kql/Kql.php#L73
-export type KirbyQueryModel =
+export type KirbyQueryModel<T extends string = never> =
   | "collection"
   | "file"
   | "kirby"
   | "page"
   | "site"
-  | "user";
+  | "user"
+  | T;
 
-export type KirbyQuery =
-  | KirbyQueryModel
-  | `${KirbyQueryModel}.${string}`
-  | `${KirbyQueryModel}(${string})`;
+export type KirbyQuery<T extends string = never> =
+  | KirbyQueryModel<T>
+  | `${KirbyQueryModel<T>}.${string}`
+  | `${KirbyQueryModel<T>}(${string})`;