diff --git a/hack/update-codegen.sh b/hack/update-codegen.sh
index 20142033b5b..ae1925f6253 100755
--- a/hack/update-codegen.sh
+++ b/hack/update-codegen.sh
@@ -40,7 +40,7 @@ echo ">> Using ${CODEGEN_PKG} for the code generator"
 chmod +x "${CODEGEN_PKG}/generate-groups.sh"
 chmod +x "${CODEGEN_PKG}/generate-internal-groups.sh"
 
-PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd -P)"
+PROJECT_ROOT=${GOPATH}/src/github.com/kubeflow/katib
 
 modules=(experiments suggestions trials common)
 versions=(v1beta1)
diff --git a/hack/update-mockgen.sh b/hack/update-mockgen.sh
index 583225b2464..38a273c480b 100755
--- a/hack/update-mockgen.sh
+++ b/hack/update-mockgen.sh
@@ -21,7 +21,7 @@ set -o errexit
 set -o nounset
 set -o pipefail
 
-SCRIPT_ROOT=${GOPATH}/src/github.com/kubeflow/katib
+SCRIPT_ROOT="$(dirname "${BASH_SOURCE[0]}")/.."
 
 cd "${SCRIPT_ROOT}"