This isn't equivalent to the previous code - I don't agree with setting up expectations in before blocks. Do you mind changing?