tests: fix tests