Update install.py

install.py

@@ -6,7 +6,7 @@ linux_packages = [
 ]
 
 linux_after_command = {
-    'zsh': 'chsh -s $(which zsh)'
+    'zsh': 'sudo chsh -s $(which zsh) $(whoami)'
 }
 
 os_type = platform.uname().system.lower()