diff --git a/lua/myworkflow/go-tests.lua b/lua/myworkflow/go-tests.lua
index 1b57428..fb62599 100644
--- a/lua/myworkflow/go-tests.lua
+++ b/lua/myworkflow/go-tests.lua
@@ -126,7 +126,7 @@ function M.goRunTests()
 end
 
 function M.goSetAllMarks()
-	local files = vim.fn.system({ "find", "-type", "f", "-name", "*.go" })
+	local files = vim.fn.system({ "find", "-type", "f", "-name", "*_test.go" })
 
 	local sep = "\n"
 	local filetable = {}