diff --git a/pkg/ast/model/table_funcs.go.tpl b/pkg/ast/model/table_funcs.go.tpl
index cbc5892..55b4e26 100644
--- a/pkg/ast/model/table_funcs.go.tpl
+++ b/pkg/ast/model/table_funcs.go.tpl
@@ -1,6 +1,11 @@
 package model
 
 import (
+	"context"
+	"time"
+
+	"{{ .PkgName }}/database/table"
+
 	. "github.com/go-jet/jet/v2/postgres"
 	log "github.com/sirupsen/logrus"
 )