Fix expectation, use equal