diff --git a/main.py b/main.py
index cbebdc1..dcccca1 100644
--- a/main.py
+++ b/main.py
@@ -39,7 +39,11 @@ def pseudo_zsh():
     username = os.getlogin()
 
     # 获取 IP 地址
-    ip_address = socket.gethostbyname(socket.gethostname())
+    try:
+        hostname = socket.gethostname()
+        ip_address = socket.gethostbyname_ex(hostname)[2][0]  # 获取第一个 IP 地址
+    except (socket.gaierror, IndexError):
+        ip_address = "127.0.0.1"  # 默认回环地址
 
     # 获取内存信息
     memory = psutil.virtual_memory()