diff --git a/cli/htmgo/go.mod b/cli/htmgo/go.mod
index 07945a3..7d67f4e 100644
--- a/cli/htmgo/go.mod
+++ b/cli/htmgo/go.mod
@@ -7,11 +7,17 @@ require (
 	github.com/google/uuid v1.6.0
 	github.com/maddalax/htmgo/framework v1.0.3-0.20241109183230-b234ead96499
 	github.com/maddalax/htmgo/tools/html-to-htmgo v0.0.0-20241109183230-b234ead96499
+	github.com/stretchr/testify v1.9.0
 	golang.org/x/mod v0.21.0
 	golang.org/x/sys v0.26.0
 	golang.org/x/tools v0.25.0
 )
 
+require (
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
+)
+
 require (
 	github.com/bmatcuk/doublestar/v4 v4.7.1
 	github.com/go-chi/chi/v5 v5.1.0 // indirect
diff --git a/cli/htmgo/tasks/astgen/project-sample/go.mod b/cli/htmgo/tasks/astgen/project-sample/go.mod
index 976e3ce..981c812 100644
--- a/cli/htmgo/tasks/astgen/project-sample/go.mod
+++ b/cli/htmgo/tasks/astgen/project-sample/go.mod
@@ -2,10 +2,12 @@ module astgen-project-sample
 
 go 1.23.0
 
-require github.com/maddalax/htmgo/framework v1.0.3-0.20241109183230-b234ead96499
+require (
+	github.com/go-chi/chi/v5 v5.1.0
+	github.com/maddalax/htmgo/framework v1.0.3-0.20241109183230-b234ead96499
+)
 
 require (
-	github.com/go-chi/chi/v5 v5.1.0 // indirect
 	github.com/google/uuid v1.6.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )