test: fix broken tests.