diff --git a/pkg/ast/model/models.gen.go.tpl b/pkg/ast/model/models.gen.go.tpl
index b5357fd..2a22f48 100644
--- a/pkg/ast/model/models.gen.go.tpl
+++ b/pkg/ast/model/models.gen.go.tpl
@@ -3,18 +3,26 @@
 // Code generated by the atomctl ; DO NOT EDIT.
 package models
 
+import (
+	"database/sql"
+)
+
+var db *sql.DB
 {{- range . }}
 var {{.PascalTable}} *{{.CamelTable}}Model
 {{- end }}
 
 // @provider(model)
 type models struct {
+	db *sql.DB
+
 {{- range . }}
 	{{.CamelTable}} *{{.CamelTable}}Model
 {{- end }}
 }
 
 func (m *models) Prepare() error {
+	db = m.db
 {{- range . }}
 	{{.PascalTable}} = m.{{.CamelTable}}
 {{- end }}