add another expectation just to be safe
broke up the single responsibility method a little bit