You can add the message in here, rather than having the two expects.