diff --git a/modules/sbt/src/main/scala/com/anymindgroup/sbt/PgCodeGenPlugin.scala b/modules/sbt/src/main/scala/com/anymindgroup/sbt/PgCodeGenPlugin.scala
index f52c1f1..8fd1d4f 100644
--- a/modules/sbt/src/main/scala/com/anymindgroup/sbt/PgCodeGenPlugin.scala
+++ b/modules/sbt/src/main/scala/com/anymindgroup/sbt/PgCodeGenPlugin.scala
@@ -59,7 +59,7 @@ object PgCodeGenPlugin extends AutoPlugin {
       pgCodeGenPassword       := None,
       pgCodeGenSqlSourceDir   := file("src") / "main" / "resources" / "db" / "migration",
       pgCodeGenOutputPackage  := "anychat.chat.db",
-      pgCodeGenOutputDir      := (Compile / sourceManaged).value,
+      pgCodeGenOutputDir      := (Compile / sourceManaged).value / "main",
       pgCodeGenExcludedTables := Nil,
       pgCodeGenUseDockerImage := Some("postgres:16-alpine"),
       pgCodeGen := {
diff --git a/modules/sbt/src/sbt-test/test/basic/test b/modules/sbt/src/sbt-test/test/basic/test
index 540736a..845c452 100644
--- a/modules/sbt/src/sbt-test/test/basic/test
+++ b/modules/sbt/src/sbt-test/test/basic/test
@@ -1,6 +1,6 @@
 > +pgCodeGen
-$ exists target/scala-2.13/src_managed/com/example/TestTable.scala
-$ exists target/scala-2.13/src_managed/com/example/TestRow.scala
-$ exists target/scala-3.3.3/src_managed/com/example/TestTable.scala
-$ exists target/scala-3.3.3/src_managed/com/example/TestRow.scala
+$ exists target/scala-2.13/src_managed/main/com/example/TestTable.scala
+$ exists target/scala-2.13/src_managed/main/com/example/TestRow.scala
+$ exists target/scala-3.3.3/src_managed/main/com/example/TestTable.scala
+$ exists target/scala-3.3.3/src_managed/main/com/example/TestRow.scala
 > +compile