diff --git a/killstream/kill_device.py b/killstream/kill_device.py index a26eec1..45f19c8 100644 --- a/killstream/kill_device.py +++ b/killstream/kill_device.py @@ -40,7 +40,7 @@ def kill_session(): exit() platform = session.players[0].platform - if DEVICES[platform]: + if DEVICES.get(platform): MESSAGE = DEVICES.get(platform, DEFAULT_REASON) print('Killing {user}\'s stream on {plat}.'.format(user=user, plat=platform)) session.stop(reason=MESSAGE)