diff --git a/go/tests/entrypoint.sh b/go/tests/entrypoint.sh
index 55bd52c1..581d3679 100755
--- a/go/tests/entrypoint.sh
+++ b/go/tests/entrypoint.sh
@@ -2,6 +2,7 @@
 
 set -e
 
+cp -r /public/go/tests .
 cp -a student piscine-go
 cd piscine-go
 
@@ -50,6 +51,7 @@ if command -v "${EXERCISE}_test" >/dev/null 2>&1; then
 	"${EXERCISE}_test"
 else
 	# The exercise is a function
-	cd "/public/go/tests/func/${EXERCISE}_test"
+	cd "$HOME/tests/func/${EXERCISE}_test"
+	go mod edit -replace "student=$HOME/piscine-go"
 	go run .
 fi