Fix expectation order bug