diff --git a/internal/env/env.go b/internal/env/env.go
index 5d602e90..3f6feec8 100644
--- a/internal/env/env.go
+++ b/internal/env/env.go
@@ -97,12 +97,12 @@ func isLLGoRoot(root string) (string, bool) {
 		return "", false
 	}
 	// Check for go.mod
-	data, err := os.ReadFile(filepath.Join(root, "go.mod"))
+	data, err := os.ReadFile(filepath.Join(root, LLGoRuntimePkgName, "go.mod"))
 	if err != nil {
 		return "", false
 	}
 	// Check module name
-	if !strings.Contains(string(data), "module "+LLGoCompilerPkg+"\n") {
+	if !strings.Contains(string(data), "module "+LLGoRuntimePkg+"\n") {
 		return "", false
 	}
 	return root, true
diff --git a/internal/env/env_test.go b/internal/env/env_test.go
index 0a0d43f2..80773cad 100644
--- a/internal/env/env_test.go
+++ b/internal/env/env_test.go
@@ -41,8 +41,10 @@ func TestLLGoRuntimeDir(t *testing.T) {
 		defer os.Setenv("LLGO_ROOT", origLLGoRoot)
 
 		tmpDir := t.TempDir()
-		goModContent := []byte("module github.com/goplus/llgo\n")
-		if err := os.WriteFile(filepath.Join(tmpDir, "go.mod"), goModContent, 0644); err != nil {
+		runtimeDir := filepath.Join(tmpDir, "runtime")
+		os.MkdirAll(runtimeDir, 0755)
+		goModContent := []byte("module github.com/goplus/llgo/runtime\n")
+		if err := os.WriteFile(filepath.Join(runtimeDir, "go.mod"), goModContent, 0644); err != nil {
 			t.Fatal(err)
 		}
 
@@ -92,8 +94,10 @@ func TestLLGoROOT(t *testing.T) {
 		defer os.Setenv("LLGO_ROOT", origLLGoRoot)
 
 		tmpDir := t.TempDir()
-		goModContent := []byte("module github.com/goplus/llgo\n")
-		if err := os.WriteFile(filepath.Join(tmpDir, "go.mod"), goModContent, 0644); err != nil {
+		runtimeDir := filepath.Join(tmpDir, "runtime")
+		os.MkdirAll(runtimeDir, 0755)
+		goModContent := []byte("module github.com/goplus/llgo/runtime\n")
+		if err := os.WriteFile(filepath.Join(runtimeDir, "go.mod"), goModContent, 0644); err != nil {
 			t.Fatal(err)
 		}
 
@@ -170,8 +174,10 @@ func TestIsLLGoRoot(t *testing.T) {
 	// Test with valid path and valid go.mod
 	t.Run("valid path and go.mod", func(t *testing.T) {
 		tmpDir := t.TempDir()
-		goModContent := []byte("module github.com/goplus/llgo\n")
-		if err := os.WriteFile(filepath.Join(tmpDir, "go.mod"), goModContent, 0644); err != nil {
+		runtimeDir := filepath.Join(tmpDir, "runtime")
+		os.MkdirAll(runtimeDir, 0755)
+		goModContent := []byte("module github.com/goplus/llgo/runtime\n")
+		if err := os.WriteFile(filepath.Join(runtimeDir, "go.mod"), goModContent, 0644); err != nil {
 			t.Fatal(err)
 		}