updated expectation tests