error while running the script

[Shameel123]

./krack_all_zero_tk.py
Traceback (most recent call last):
File "./krack_all_zero_tk.py", line 10, in
username = os.path.expanduser(os.environ["SUDO_USER"])
File "/usr/lib/python2.7/UserDict.py", line 40, in getitem
raise KeyError(key)
KeyError: 'SUDO_USER'

[Shameel123]

solved it myself.
By changing "SUDO_USER" to "USER"