Fix test expectation