diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
index bbb476f9..6cc6ffe3 100644
--- a/pyrogram/client/client.py
+++ b/pyrogram/client/client.py
@@ -3400,7 +3400,7 @@ class Client:
             return False
 
     def download_media(self,
-                       message: pyrogram_types.Message,
+                       message: pyrogram_types.Message or str,
                        file_name: str = "",
                        block: bool = True,
                        progress: callable = None):