diff --git a/main.go b/main.go
index 3e22b7ae..83aa09cb 100644
--- a/main.go
+++ b/main.go
@@ -17,7 +17,8 @@ import (
 	"x-ui/web/service"
 
 	"github.com/op/go-logging"
-	"github.com/shirou/gopsuti"
+	"github.com/shirou/gopsuti/cpu"
+	"github.com/shirou/gopsuti/disk"
 )
 
 func runWebServer() {