Let wait_for_ack take a block