diff --git a/src/backend/lib/display.py b/src/backend/lib/display.py index 1435fed..183589f 100755 --- a/src/backend/lib/display.py +++ b/src/backend/lib/display.py @@ -11,8 +11,10 @@ class TerminalDisplay: self.term = True self.w, self.y = os.get_terminal_size()[0], os.get_terminal_size()[1] self.home = os.environ["HOME"] - breakpoint() - self.user = os.environ["USER"] + try: + self.user = os.environ["USER"] + except KeyError: + self.user = os.environ["PWD"] self.version = "0.4.0" self.slogan = "The Installer Initiative" self.green = "\033[1;32m"