still need to find right expectation for model