diff --git a/emailtunnel/__init__.py b/emailtunnel/__init__.py index 437a6c5..202d7ee 100644 --- a/emailtunnel/__init__.py +++ b/emailtunnel/__init__.py @@ -83,7 +83,10 @@ def __init__(self, message=None): if message: assert isinstance(message, bytes) - self.message = email.message_from_bytes(message) + if six.PY3: + self.message = email.message_from_bytes(message) + else: + self.message = email.message_from_string(message) if not self._sanity_check(message): self._sanity_log_invalid(message)