I guess the solution is configuring your power button to enter sleep? That works for Wayland as well. Just don't long-press it to open the power menu...