diff --git a/cmd_gen/main.go b/cmd_gen/main.go
index 12e1c70..8ae91cc 100644
--- a/cmd_gen/main.go
+++ b/cmd_gen/main.go
@@ -237,7 +237,7 @@ function run-dir() {
   if [[ $status -eq "0" ]]; then
     go mod init &>> ${prefix}pass || status=1
     go mod tidy &>> ${prefix}pass || status=1
-    goimports -w . &>> ${prefix}pass || status=1
+    goimports -w *.go &>> ${prefix}pass || status=1
     go build &>> ${prefix}pass || status=1
   fi
   if [[ $status -eq "1" ]]; then
diff --git a/cmd_gen/main_test.go b/cmd_gen/main_test.go
index 595a13b..94c4480 100644
--- a/cmd_gen/main_test.go
+++ b/cmd_gen/main_test.go
@@ -207,7 +207,7 @@ function run-dir() {
   if [[ $status -eq "0" ]]; then
     go mod init &>> ${prefix}pass || status=1
     go mod tidy &>> ${prefix}pass || status=1
-    goimports -w . &>> ${prefix}pass || status=1
+    goimports -w *.go &>> ${prefix}pass || status=1
     go build &>> ${prefix}pass || status=1
   fi
   if [[ $status -eq "1" ]]; then