Fix compilation failure for tests