Could you do this with before / after blocks instead of ensures? I think that would also clean up the next spec a lot if you also split it into two specs with the same before / after block.