Fixed tests build breaking.