Change expectation to stub