diff --git a/src/extendWithZodSchema.ts b/src/extendWithZodSchema.ts
index 3323a94..7dd3f1f 100644
--- a/src/extendWithZodSchema.ts
+++ b/src/extendWithZodSchema.ts
@@ -66,13 +66,10 @@ export function extendWithZodSchema<
         SchemaValues
     >,
 ) {
-
-    const class_ = class extends extend {
+    return class extends extend {
         static readonly schema = schema
-        readonly schema = schema
-    }
-
-    return class_ as unknown as (
+               readonly schema = schema
+    } as unknown as (
         Class<
             Omit<C, "schema"> &
             { readonly schema: typeof schema } &
@@ -83,7 +80,6 @@ export function extendWithZodSchema<
         Omit<StaticMembers<C>, "schema"> &
         { readonly schema: typeof schema }
     )
-
 }
 
 
@@ -99,4 +95,4 @@ const Test2Schema = Test1.schema.extend({ prout: z.literal("ruquier"), ruquier:
 const Test2 = extendWithZodSchema(Test1, Test2Schema)
 
 Test2.schema
-new Test2().schema
+new Test2().prout