test: Adapt to previous changes.

Fixes #696