Move checkExpectation from then to fin block.