diff --git a/cmd/generate.go b/cmd/generate.go
index 6545d16..4692586 100644
--- a/cmd/generate.go
+++ b/cmd/generate.go
@@ -76,8 +76,18 @@ func generate(scheamPath, outDir string) error {
 			return err
 		}
 	}
+
 	sb.WriteString(")")
 
+	sb.WriteString(`
+	func DerivedTable(tblName string, fromQry pgm.Query) pgm.Table {
+		t := pgm.Table{
+			Name:         tblName,
+			DerivedTable: fromQry,
+		}
+		return t
+}`)
+
 	// Format code before saving
 	code, err := formatGoCode(sb.String())
 	if err != nil {
diff --git a/playground/db/derived.go b/playground/db/derived.go
deleted file mode 100644
index 4fcbf1d..0000000
--- a/playground/db/derived.go
+++ /dev/null
@@ -1,11 +0,0 @@
-package db
-
-import "code.patial.tech/go/pgm"
-
-func DerivedTable(tblName string, fromQry pgm.Query) pgm.Table {
-	t := pgm.Table{
-		Name:         tblName,
-		DerivedTable: fromQry,
-	}
-	return t
-}
diff --git a/playground/db/schema.go b/playground/db/schema.go
index 7e641e2..9446337 100644
--- a/playground/db/schema.go
+++ b/playground/db/schema.go
@@ -12,3 +12,11 @@ var (
 	Comment     = pgm.Table{Name: "comments", FieldCount: 5}
 	Employee    = pgm.Table{Name: "employees", FieldCount: 5}
 )
+
+func DerivedTable(tblName string, fromQry pgm.Query) pgm.Table {
+	t := pgm.Table{
+		Name:         tblName,
+		DerivedTable: fromQry,
+	}
+	return t
+}