I think it would be better to wait for `_back_button` to be not displayed as it's the last thing to leave the viewport whereas edit is the first.