diff --git a/data/nvim/init.lua b/data/nvim/init.lua
index e07c209..f2ed3d8 100644
--- a/data/nvim/init.lua
+++ b/data/nvim/init.lua
@@ -341,8 +341,12 @@ require("lazy").setup({
 						alternate = "{}_test.go",
 						type = "source",
 					},
+					["*_generated.go"] = {
+						alternate = "{}_test.go",
+						type = "source",
+					},
 					["*_test.go"] = {
-						alternate = "{}.go",
+						alternate = { "{}.go", "{}_generated.go" },
 						type = "test",
 					},
 				},