diff --git a/packages/runtime/src/client/contract.ts b/packages/runtime/src/client/contract.ts
index 4206159f..94247354 100644
--- a/packages/runtime/src/client/contract.ts
+++ b/packages/runtime/src/client/contract.ts
@@ -664,7 +664,7 @@ export type AllModelOperations<Schema extends SchemaDef, Model extends GetModels
      */
     delete<T extends DeleteArgs<Schema, Model>>(
         args: SelectSubset<T, DeleteArgs<Schema, Model>>,
-    ): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model>>>;
+    ): ZenStackPromise<Schema, Simplify<ModelResult<Schema, Model, T>>>;
 
     /**
      * Deletes multiple entities.
diff --git a/packages/runtime/test/plugin/entity-mutation-hooks.test.ts b/packages/runtime/test/plugin/entity-mutation-hooks.test.ts
index be5d8edf..6ca7f112 100644
--- a/packages/runtime/test/plugin/entity-mutation-hooks.test.ts
+++ b/packages/runtime/test/plugin/entity-mutation-hooks.test.ts
@@ -235,7 +235,7 @@ describe('Entity mutation hooks tests', () => {
                                     name: 'A user',
                                 }),
                                 expect.objectContaining({
-                                    email: 'u1@test.com',
+                                    email: 'u2@test.com',
                                     name: 'A user',
                                 }),
                             ]),
diff --git a/packages/runtime/test/plugin/on-query-hooks.test.ts b/packages/runtime/test/plugin/on-query-hooks.test.ts
index 42e19ec0..5cb72c97 100644
--- a/packages/runtime/test/plugin/on-query-hooks.test.ts
+++ b/packages/runtime/test/plugin/on-query-hooks.test.ts
@@ -1,6 +1,6 @@
 import SQLite from 'better-sqlite3';
 import { SqliteDialect } from 'kysely';
-import { beforeEach, describe, expect, it } from 'vitest';
+import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { definePlugin, ZenStackClient, type ClientContract } from '../../src/client';
 import { schema } from '../schemas/basic';
 
@@ -14,6 +14,10 @@ describe('On query hooks tests', () => {
         await _client.$pushSchema();
     });
 
+    afterEach(async () => {
+        await _client?.$disconnect();
+    });
+
     it('supports simple interception', async () => {
         const user = await _client.user.create({
             data: { email: 'u1@test.com' },