diff --git a/lib/sshyp.py b/lib/sshyp.py index 301fac5..bae1d0d 100755 --- a/lib/sshyp.py +++ b/lib/sshyp.py @@ -802,6 +802,7 @@ if __name__ == "__main__": success_flag = True read_shortcut() elif arguments[0] == 'tweak': + success_flag = True from stweak import global_menu global_menu(False) @@ -819,6 +820,7 @@ if __name__ == "__main__": success_flag = True whitelist_setup() elif arg_count == 1 and arguments[0] == 'tweak': + success_flag = True from stweak import global_menu global_menu(False) elif arg_count > 2 and arguments[1] in ('add', 'del'): diff --git a/lib/stweak.py b/lib/stweak.py index 00a1e36..abdd735 100644 --- a/lib/stweak.py +++ b/lib/stweak.py @@ -88,7 +88,7 @@ def install_type(): _dev_type = 'server' Path(f"{home}/.config/sshyp/deleted").mkdir(mode=0o700, exist_ok=True) Path(f"{home}/.config/sshyp/whitelist").mkdir(mode=0o700, exist_ok=True) - curses_terminate('\nmake sure the ssh service is running and properly configured') + curses_terminate('\nmake sure the ssh service is running and properly configured\n') else: _dev_type = 'client' if _install_type == 2: