diff --git a/trafficCog/__init__.py b/trafficCog/__init__.py
index 27b696d..d1af273 100644
--- a/trafficCog/__init__.py
+++ b/trafficCog/__init__.py
@@ -1,6 +1,6 @@
 from redbot.core.bot import Red
 from .trafficCog import TrafficCog
 
-def setup(bot):
+async def setup(bot: Red):
     cog = TrafficCog(bot)
-    bot.add_cog(cog)
+    await bot.add_cog(cog)
\ No newline at end of file
diff --git a/trafficCog/trafficCog.py b/trafficCog/trafficCog.py
index 7b0ad36..29a3eb6 100644
--- a/trafficCog/trafficCog.py
+++ b/trafficCog/trafficCog.py
@@ -92,4 +92,5 @@ class TrafficCog(commands.Cog):
                 await channel.send(f"{member.display_name} has been banned from the server.")
 
 def setup(bot):
-    bot.add_cog(TrafficCog(bot))
+    cog = TrafficCog(bot)
+    bot.add_cog(cog)
\ No newline at end of file