diff --git a/shortmute/shortmute.py b/shortmute/shortmute.py
index 9add5e2..53c253e 100644
--- a/shortmute/shortmute.py
+++ b/shortmute/shortmute.py
@@ -17,7 +17,7 @@ class Shortmute(commands.Cog):
 
     @app_commands.command()
     @app_commands.rename(target='member')
-    async def shortmute(self, interaction: discord.Interaction, target: discord.Member, duration: int, reason: str, evidence_link: str = None, evidence_image: discord.AppCommandOptionType.attachment = None):
+    async def shortmute(self, interaction: discord.Interaction, target: discord.Member, duration: int, reason: str, evidence_link: str = None, evidence_image: discord.Attachment = None):
         """Shortmute someone for up to 30m.
 
         Parameters
@@ -30,7 +30,7 @@ class Shortmute(commands.Cog):
             The reason for the shortmute
         evidence_link: str = None
             An image link to evidence for the shortmute, do not use with evidence_image
-        evidence_image: discord.AppCommandOptionType.attachment = None
+        evidence_image: discord.Attachment = None
             An image file used as evidence for the shortmute, do not use with evidence_link
             """
         passed_info = {