diff --git a/reginaldCog/llm_clients/openai_data_models.py b/reginaldCog/llm_clients/openai_data_models.py
index 21ab7c8..171e7c4 100644
--- a/reginaldCog/llm_clients/openai_data_models.py
+++ b/reginaldCog/llm_clients/openai_data_models.py
@@ -23,8 +23,8 @@ class OutputText(Content):
 @dataclass(frozen=True)
 class InputImage(Content):
     type: str = field(init=False, default='input_image')
-    image_url: str = field(default=None)
-    file_id: str = field(default=None)
+    image_url: str | None = field(default=None)
+    file_id: str | None = field(default=None)
 
     def __post_init__(self):
         if self.image_url is None and self.file_id is None:
diff --git a/reginaldCog/messenger_clients/discord_data_models.py b/reginaldCog/messenger_clients/discord_data_models.py
index 8abb41e..56f621b 100644
--- a/reginaldCog/messenger_clients/discord_data_models.py
+++ b/reginaldCog/messenger_clients/discord_data_models.py
@@ -1,4 +1,4 @@
-from dataclasses import dataclass, field, asdict
+from dataclasses import dataclass, field
 from datetime import datetime
 
 
diff --git a/reginaldCog/messenger_clients/messenger_client.py b/reginaldCog/messenger_clients/messenger_client.py
index cfc0b5e..685c898 100644
--- a/reginaldCog/messenger_clients/messenger_client.py
+++ b/reginaldCog/messenger_clients/messenger_client.py
@@ -78,6 +78,7 @@ if __name__ == "__main__":
     class TestMessage:
         content = 'Hello!'
         author = TestAuthor()
+        attachments = []
 
 
     message = DiscordMessageBuilder(raw_message=TestMessage).create_message()