diff --git a/entrypoint.sh b/entrypoint.sh
index 52521485..93abb19b 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -11,7 +11,7 @@ cp -r student piscine-go
 cd piscine-go
 
 if test "$CODE_EDITOR_MODE"; then
-	go mod init main 2>/dev/null
+	go mod init piscine 2>/dev/null
 	GOSUMDB=off go get github.com/01-edu/z01@v0.1.0 2>/dev/null
 fi
 
@@ -24,7 +24,6 @@ if test "$CODE_EDITOR_RUN_ONLY" = true; then
 	else
 		# The exercise is a function
 		go run . "$@"
-		fi
 	fi
 	exit
 fi