nit: `stub.yields` is your friend :)