diff --git a/zhihu/auto_login.py b/zhihu/auto_login.py
index 0fca5c1..806034a 100644
--- a/zhihu/auto_login.py
+++ b/zhihu/auto_login.py
@@ -67,6 +67,7 @@ def login(email, password):
     login_code = response.json()
     print(login_code['msg'])
     print(session.cookies)
+    session.cookies.save()
     r = session.get("https://www.zhihu.com/settings/profile", headers=headers)
     print(r.status_code)
     print(r.text)
@@ -74,8 +75,6 @@ def login(email, password):
         f.write(r.content)
 
 
-if __name__ == '__main__':
-
 if __name__ == '__main__':
     email = "xxxx"
     password = "xxxxx"