diff --git a/crypto/aes/aes.go b/crypto/aes/aes.go
index bd848cd..2decd44 100644
--- a/crypto/aes/aes.go
+++ b/crypto/aes/aes.go
@@ -153,7 +153,7 @@ func generateKey(key []byte) (genKey []byte) {
 
 func AesKeyCheck(key string) (string, error) {
 	// 从环境变量获取密钥
-	keyHex := os.Getenv("RST_KEY")
+	keyHex := os.Getenv(key)
 	if keyHex == "" {
 		fmt.Println("环境变量 RST_KEY 未设置")
 		return "", errors.New("环境变量 RST_KEY 未设置")