diff --git a/omemo/omemo/state.py b/omemo/omemo/state.py index 4086635..050db17 100644 --- a/omemo/omemo/state.py +++ b/omemo/omemo/state.py @@ -211,9 +211,12 @@ class OmemoState: log.warning('Duplicate message found ' + str(e.args)) return - result = decrypt(key, iv, payload) - - log.debug("Decrypted Message => " + result) + if payload is None: + result = None + log.debug("Decrypted Key Exchange Message") + else: + result = decrypt(key, iv, payload) + log.debug("Decrypted Message => " + result) return result def create_msg(self, from_jid, jid, plaintext):