diff --git a/scripts/update-yamlfmt.sh b/scripts/update-yamlfmt.sh
index ecd46561c..90ec8aa62 100755
--- a/scripts/update-yamlfmt.sh
+++ b/scripts/update-yamlfmt.sh
@@ -40,4 +40,4 @@ find_files() {
 }
 
 export GO111MODULE=on
-find_files | xargs go run tools/yamlfmt/main.go
\ No newline at end of file
+find_files | xargs go run tools/yamlfmt/yamlfmt.go
\ No newline at end of file
diff --git a/test/typecheck/typecheck.go b/test/typecheck/typecheck.go
index 235be95cd..8f353881b 100644
--- a/test/typecheck/typecheck.go
+++ b/test/typecheck/typecheck.go
@@ -75,7 +75,7 @@ var (
 		"pkg/kubectl/cmd/testdata/edit",
 		// Tools we use for maintaining the code base but not necessarily
 		// ship as part of the release
-		"sopenim::golang::setup_env:tools",
+		"sopenim::golang::setup_env:tools/yamlfmt/yamlfmt.go:tools",
 	}
 )