Fix broken expectation