Fixed bug in contain expectation