Fix for compilation of tests