diff --git a/cli/htmgo/tasks/astgen/entry.go b/cli/htmgo/tasks/astgen/entry.go
index 6b5795a..fc86b59 100644
--- a/cli/htmgo/tasks/astgen/entry.go
+++ b/cli/htmgo/tasks/astgen/entry.go
@@ -233,10 +233,6 @@ func findPublicFuncsReturningHPage(dir string) ([]Page, error) {
 }
 
 func buildGetPartialFromContext(builder *CodeBuilder, partials []Partial) {
-	if len(partials) == 0 {
-		return
-	}
-
 	moduleName := GetModuleName()
 
 	var routerHandlerMethod = func(path string, caller string) string {
@@ -290,10 +286,13 @@ func writePartialsFile() {
 	builder := NewCodeBuilder(nil)
 	builder.AppendLine(GeneratedFileLine)
 	builder.AppendLine(PackageName)
-	builder.AddImport(ModuleName)
-	builder.AddImport(HttpModuleName)
 	builder.AddImport(ChiModuleName)
 
+	if len(partials) > 0 {
+		builder.AddImport(ModuleName)
+		builder.AddImport(HttpModuleName)
+	}
+
 	moduleName := GetModuleName()
 	for _, partial := range partials {
 		builder.AddImport(fmt.Sprintf(`%s/%s`, moduleName, partial.Import))