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