def override_handle_status()

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