diff --git a/scripts/update-yamlfmt.sh b/scripts/update-yamlfmt.sh
index 6a451c819..ecd46561c 100755
--- a/scripts/update-yamlfmt.sh
+++ b/scripts/update-yamlfmt.sh
@@ -21,7 +21,7 @@ set -o pipefail
 OPENIM_ROOT=$(dirname "${BASH_SOURCE[0]}")/..
 source "${OPENIM_ROOT}/scripts/lib/init.sh"
 
-kube::golang::setup_env
+openim::golang::setup_env
 
 cd "${OPENIM_ROOT}"
 
diff --git a/test/typecheck/typecheck.go b/test/typecheck/typecheck.go
index 88b245141..235be95cd 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
-		"scripts/tools",
+		"sopenim::golang::setup_env:tools",
 	}
 )