test: Adapt for previous changes.