Another test expectation fix