Fix the tests build