Fix tp_poll_wait to work with sgen