I think you can use $expectation in this expression rather than reinstantiating