diff --git a/src/ExtendZodSchemaClass.ts b/src/ExtendZodSchemaClass.ts
index 05143de..e65ee3d 100644
--- a/src/ExtendZodSchemaClass.ts
+++ b/src/ExtendZodSchemaClass.ts
@@ -1,7 +1,7 @@
 import { TraitInstance, TraitStaticMembers, expression } from "@thilawyn/traitify-ts"
 import { AbstractClass } from "type-fest"
 import { z } from "zod"
-import { ZodSchemaObject, ZodSchemaObjectTrait } from "./lib"
+import { ZodSchemaObject, ZodSchemaObjectTrait } from "./traits/ZodSchemaObject"
 import { Extend, StaticMembers } from "./util"