Updated test expectation