diff --git a/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyInline.scala b/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyInline.scala
index 524fc5c9e4..18592aa56a 100644
--- a/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyInline.scala
+++ b/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyInline.scala
@@ -1,6 +1,7 @@
 package test.api.v1
 
 import test.component._
+import zio.schema._
 
 object Users {
   import zio.http._
diff --git a/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyInlineNested.scala b/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyInlineNested.scala
index cc2dce0223..03a94ca498 100644
--- a/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyInlineNested.scala
+++ b/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyInlineNested.scala
@@ -1,6 +1,7 @@
 package test.api.v1
 
 import test.component._
+import zio.schema._
 
 object Users {
   import zio.http._
diff --git a/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyWithKeywordsInline.scala b/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyWithKeywordsInline.scala
index 496f1c9284..0cfa8c3dd3 100644
--- a/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyWithKeywordsInline.scala
+++ b/zio-http-gen/src/test/resources/EndpointWithRequestResponseBodyWithKeywordsInline.scala
@@ -1,6 +1,7 @@
 package test.api.v1
 
 import test.component._
+import zio.schema._
 
 object Keywords {
   import zio.http._
diff --git a/zio-http-gen/src/test/scala/zio/http/gen/openapi/EndpointGenSpec.scala b/zio-http-gen/src/test/scala/zio/http/gen/openapi/EndpointGenSpec.scala
index b0a4bd028d..e445fd7a17 100644
--- a/zio-http-gen/src/test/scala/zio/http/gen/openapi/EndpointGenSpec.scala
+++ b/zio-http-gen/src/test/scala/zio/http/gen/openapi/EndpointGenSpec.scala
@@ -910,7 +910,7 @@ object EndpointGenSpec extends ZIOSpecDefault {
           val expected = Code.File(
             List("api", "v1", "Users.scala"),
             pkgPath = List("api", "v1"),
-            imports = List(Code.Import.FromBase(path = "component._")),
+            imports = List(Code.Import.FromBase(path = "component._"), Code.Import.Absolute("zio.schema._")),
             objects = List(
               Code.Object(
                 "Users",
@@ -1026,7 +1026,7 @@ object EndpointGenSpec extends ZIOSpecDefault {
           val expected = Code.File(
             List("api", "v1", "Users.scala"),
             pkgPath = List("api", "v1"),
-            imports = List(Code.Import.FromBase(path = "component._")),
+            imports = List(Code.Import.FromBase(path = "component._"), Code.Import.Absolute("zio.schema._")),
             objects = List(
               Code.Object(
                 "Users",