fixed Message.wait_for

Author: kvsxxx

README.md

@@ -172,6 +172,15 @@ You can find more (and better) examples [here](https://github.com/discord-py-ui/
 
 # Changelog
 
+-   <details>
+    <summary>4.0.3</summary>
+    
+    ## **Fixed**
+    - Message.wait_for
+
+
+    </details>
+
 -   <details>
     <summary>4.0.2</summary>
 

discord_ui/__init__.py

@@ -44,4 +44,4 @@
 
 
 __title__ = "discord-ui"
-__version__ = "4.0.2"
+__version__ = "4.0.3"

discord_ui/receive.py

@@ -563,8 +563,8 @@ async def wait_for(self, event_name: typing.Literal["select", "button"], client,
         :type: :class:`~PressedButton` | :class:`~SelectedMenu`
         """
         if event_name.lower() in ["button", "select"]:
-            def _check(com, msg):
-                if msg.id == self.id:
+            def _check(com):
+                if com.message.id == self.id:
                     statements = []
                     if custom_id is not MISSING:
                         statements.append(com.custom_id == custom_id)