yeah changing `return`s to `yield`s works, i think this is fixable on the rust side actually