diff --git a/aurora/aurora.py b/aurora/aurora.py
index 2674151..2f9c7d1 100644
--- a/aurora/aurora.py
+++ b/aurora/aurora.py
@@ -68,10 +68,10 @@ class Aurora(commands.Cog):
         self.bot = bot
         register_config(config)
         disable_dateutil()
-        self.handle_expiry.start() # pylint: disable=no-member
 
     async def cog_load(self):
-        """This method prepares the database schema for all of the guilds the bot is currently in."""
+        """This method prepares the database schema for all of the guilds the bot is currently in, and starts the handle_expiry task."""
+        await self.handle_expiry.start() # pylint: disable=no-member
         guilds: list[discord.Guild] = self.bot.guilds
 
         try:
@@ -83,7 +83,7 @@ class Aurora(commands.Cog):
             return
 
     async def cog_unload(self):
-        self.handle_expiry.cancel() # pylint: disable=no-member
+        await self.handle_expiry.cancel() # pylint: disable=no-member
 
     @commands.Cog.listener('on_guild_join')
     async def db_generate_guild_join(self, guild: discord.Guild):
@@ -451,7 +451,7 @@ class Aurora(commands.Cog):
             await interaction.response.send_message(content=f"{target.mention} has been banned for {humanize.precisedelta(parsed_time)}!\n**Reason** - `{reason}`")
 
             try:
-                embed = await embed_factory('message', await self.bot.get_embed_color(None), guild=interaction.guild, moderator=interaction.user, reason=reason, moderation_type='tempbanned', response=await interaction.original_response(), duration=parsed_time)
+                embed = await embed_factory('message', await self.bot.get_embed_color(None) , guild=interaction.guild, moderator=interaction.user, reason=reason, moderation_type='tempbanned', response=await interaction.original_response(), duration=parsed_time)
                 await target.send(embed=embed)
             except discord.errors.HTTPException:
                 pass
diff --git a/aurora/utilities/embed_factory.py b/aurora/utilities/embed_factory.py
index 2c97274..7f55f1f 100644
--- a/aurora/utilities/embed_factory.py
+++ b/aurora/utilities/embed_factory.py
@@ -20,6 +20,7 @@ async def embed_factory(embed_type: str, color: Color, /,  interaction: Interact
     - 'changes'
 
     Required arguments for 'message':
+    - color
     - guild
     - reason
     - moderation_type