[test] Fix tests