I changed it, but I went with the `yield_control` from rspec