in main.py [0:0]
def override_handle_status(self, key ,value):
save = self.key_id # in case we need to restore it
handle(self, key, value) # call original code
if key in ('UNEXPECTED', 'FAILURE'):
self.key_id = save # restore the overwritten value