Fix compilation of tests