Fixed tests build
Test fix