Fix compilation of the test