diff --git a/tests/e2e/mongoose.spec.ts b/tests/e2e/mongoose.spec.ts
index f6ca8436..20daeac4 100644
--- a/tests/e2e/mongoose.spec.ts
+++ b/tests/e2e/mongoose.spec.ts
@@ -58,7 +58,7 @@ describe('Mongoose', () => {
       ...createDto,
       name: 'Nest',
       age: 5,
-      kitten: [kitten._id as string],
+      kitten: [kitten._id.toString()],
     };
 
     const parent = await new Promise<string>((resolve) => {
diff --git a/tests/src/cats/schemas/cat.schema.ts b/tests/src/cats/schemas/cat.schema.ts
index 72e2c735..bcd0e243 100644
--- a/tests/src/cats/schemas/cat.schema.ts
+++ b/tests/src/cats/schemas/cat.schema.ts
@@ -2,7 +2,7 @@ import { Document, Types } from 'mongoose';
 import { Prop, Schema, SchemaFactory } from '../../../../lib';
 
 @Schema()
-export class Cat extends Document {
+export class Cat extends Document<Types.ObjectId> {
   @Prop()
   name: string;