tests: fix tests.