add expectation to test