moved expectation to bottom of test