[test] Fix broken tests