From 1a6bd44c055739752a6d6f7a7131cff7ca30ce35 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Sun, 25 Feb 2024 04:49:04 +0100
Subject: [PATCH] Extend work

---
 src/builders/ZodSchemaClassExtender.ts |  2 +-
 src/tests.ts                           | 11 +++--------
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/src/builders/ZodSchemaClassExtender.ts b/src/builders/ZodSchemaClassExtender.ts
index 24331a8..f2233bf 100644
--- a/src/builders/ZodSchemaClassExtender.ts
+++ b/src/builders/ZodSchemaClassExtender.ts
@@ -47,7 +47,7 @@ export class ZodSchemaClassExtender<Superclass extends AbstractClass<object>> {
 
         return new this.constructor(
             this.superclass as unknown as AbstractClass<
-                Extend<[SuperInstance, Schemas]>,
+                Extend<[SuperInstance, Values]>,
                 [values: Values]
             > &
             Extend<[
diff --git a/src/tests.ts b/src/tests.ts
index dd2cecd..9498792 100644
--- a/src/tests.ts
+++ b/src/tests.ts
@@ -41,21 +41,16 @@ const jsonifiedUser = await inst.jsonifyPromise()
 const extended = User.extend()
     .schema({
         schema: s => s.extend({
-            name: z.literal("Admin")
+            prout: z.string()
         }),
         defaultValues: v => v,
     })
 
 
-class SubTest extends User.extend()
-    .schema({
-        schema: s => s.extend({}),
-        defaultValues: v => v,
-    })
-    .build()
-{}
+class SubTest extends extended.build() {}
 
 const subInst = await SubTest.createPromise({ name: "", prout: "" })
+
 // console.log(subInst)