diff --git a/Dockerfile b/Dockerfile
index f68f35d..a95d770 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,7 +6,7 @@ COPY go.mod go.sum ./
 RUN go env -w GO111MODULE=on
 RUN go env -w GOPROXY=https://goproxy.cn,direct
 RUN go mod download
-COPY main.go .env ./
+COPY main.go ./
 
 RUN go build -o api
 
diff --git a/main.go b/main.go
index 95561e6..cbbd4f6 100644
--- a/main.go
+++ b/main.go
@@ -12,7 +12,6 @@ import (
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
 	"github.com/glebarez/sqlite"
-	"github.com/joho/godotenv"
 	"github.com/openai/openai-go/v2"
 	"github.com/openai/openai-go/v2/option"
 	"gorm.io/gorm"
@@ -31,11 +30,6 @@ type PresetCodeInput struct {
 }
 
 func main() {
-	err := godotenv.Load()
-	if err != nil {
-		log.Fatal(err)
-	}
-
 	db, err := gorm.Open(sqlite.Open("database.db"), &gorm.Config{})
 
 	if err != nil {
@@ -113,9 +107,8 @@ func main() {
 		systemPrompt := "你是编程老师，擅长分析代码和错误信息，一般出错在语法和格式，请指出错误在第几行，并给出中文的、简要的解决方法。用 markdown 格式返回。"
 		userPrompt := fmt.Sprintf("编程语言：%s\n代码：\n```%s\n```\n错误信息：\n```%s\n```", payload.Language, payload.Code, payload.ErrorInfo)
 
-		apiKey := os.Getenv("API_KEY")
-
-		if apiKey == "" {
+		apiKey, ok := os.LookupEnv("API_KEY")
+		if !ok || apiKey == "" {
 			c.JSON(http.StatusBadRequest, gin.H{"error": "API_KEY is not set"})
 			return
 		}