diff --git a/database/drivers/mysql/parse.go b/database/drivers/mysql/parse.go
index 26f6528..1e59c13 100644
--- a/database/drivers/mysql/parse.go
+++ b/database/drivers/mysql/parse.go
@@ -15,6 +15,8 @@ import (
 	"gnorm.org/gnorm/database/drivers/mysql/gnorm/tables"
 )
 
+//go:generate gnorm gen
+
 // MySQL implements drivers.Driver interface for MySQL database.
 type MySQL struct{}
 
diff --git a/database/drivers/mysql/templates/table.gotmpl b/database/drivers/mysql/templates/table.gotmpl
index cb536c2..0257099 100644
--- a/database/drivers/mysql/templates/table.gotmpl
+++ b/database/drivers/mysql/templates/table.gotmpl
@@ -5,7 +5,7 @@ package {{toLower .Name}}
 import "gnorm.org/gnorm/database/drivers/mysql/gnorm"
 
 {{$table := .DBName -}}
-{{$schema := .DBSchema -}}
+{{$schema := .Schema.DBName -}}
 // Row represents a row from '{{ $table }}'.
 type Row struct {
 {{- range .Columns }}
diff --git a/database/drivers/postgres/parse.go b/database/drivers/postgres/parse.go
index 760d652..553262f 100644
--- a/database/drivers/postgres/parse.go
+++ b/database/drivers/postgres/parse.go
@@ -15,6 +15,8 @@ import (
 	"gnorm.org/gnorm/database/drivers/postgres/gnorm/tables"
 )
 
+//go:generate gnorm gen
+
 // PG implements drivers.Driver interface for interacting with postgresql
 // database.
 type PG struct{}