test: Fix tests.