diff --git a/main.go b/main.go
index 1b2e2ba..e73844d 100644
--- a/main.go
+++ b/main.go
@@ -1,7 +1,6 @@
 package main
 
 import (
-	"bytes"
 	"encoding/json"
 	"flag"
 	"fmt"
@@ -18,19 +17,17 @@ import (
 )
 
 var (
-	configFilePath string
+	configFile     string
 	sprayEquipment = equipment.NewSprayEquipment("DC73AEE4CD274074BD9D01BCEE379A98")
 )
 
 func init() {
-	flag.StringVar(&configFilePath, "config", "config.yml", "配置文件")
+	flag.StringVar(&configFile, "config", "config.yml", "配置文件")
+	flag.Parse()
 
-	buf, err := os.ReadFile(configFilePath)
-	if err != nil {
-		panic(err)
-	}
-	viper.SetConfigType("yaml")
-	if err := viper.ReadConfig(bytes.NewBuffer(buf)); err != nil {
+	fmt.Println(configFile)
+	viper.SetConfigFile(configFile)
+	if err := viper.ReadInConfig(); err != nil {
 		panic(err)
 	}