tests: fix broken tests