diff --git a/internal/ui/windows/gen.go b/internal/ui/windows/gen.go
index 12882d7..de70617 100644
--- a/internal/ui/windows/gen.go
+++ b/internal/ui/windows/gen.go
@@ -1,3 +1,3 @@
 package windows
 
-//go:generate go run -mod=mod github.com/icedream/ui2walk
+//go:generate go run -mod=mod github.com/icedream/ui2walk -pkg $GOPACKAGE