diff --git a/src/data.py b/src/data.py
index 041589b..1711cb0 100644
--- a/src/data.py
+++ b/src/data.py
@@ -1,4 +1,5 @@
 from enum import Enum
+from typing import Union
 import re
 import datetime
 import logging
@@ -87,8 +88,8 @@ class Message():
                  date: datetime.datetime,
                  user: str,
                  channel: Channel,
-                 guild: str = None,
-                 to_from: str = None) -> None:
+                 guild: Union[str, None] = None,
+                 to_from: Union[str, None] = None) -> None:
         self.message = message
         self.date = date
         self.channel = channel