diff --git a/dist/lib/index.d.ts b/dist/lib/index.d.ts
index 9dc702f..d0c25f0 100644
--- a/dist/lib/index.d.ts
+++ b/dist/lib/index.d.ts
@@ -70,7 +70,7 @@ export declare type Terms<T> = {
  * A `SearchFn` function matches documents in a single collection and returns a `SearchResult` based on the given
  * `terms`, `limit`, and `sort`.
  */
-export declare type SearchFn<T extends Searchable, S extends Searchable = T> = (terms?: Terms<Document<DeepNonNullable<S>>>, limit?: Limit, sort?: Sort<Document<T>>[] | Sort<Document<T>>) => Promise<SearchResult<T>>;
+export declare type SearchFn<T extends Searchable, S extends Searchable = T> = (terms?: Terms<Document<DeepNonNullable<S>>>, limit?: Limit, sort?: Sort<Document<S>>[] | Sort<Document<S>>) => Promise<SearchResult<T>>;
 /**
  * Search results are a tuple of three values:
  *   1. The **total** number of matching documents in the searched collection, ignoring limit
diff --git a/lib/index.ts b/lib/index.ts
index b80cb04..0727553 100644
--- a/lib/index.ts
+++ b/lib/index.ts
@@ -85,7 +85,7 @@ export type Terms<T> = {
 export type SearchFn<T extends Searchable, S extends Searchable = T> = (
   terms?: Terms<Document<DeepNonNullable<S>>>,
   limit?: Limit,
-  sort?: Sort<Document<T>>[] | Sort<Document<T>>
+  sort?: Sort<Document<S>>[] | Sort<Document<S>>
 ) => Promise<SearchResult<T>>
 
 /**