[test] Fix build