Fix compilation of tests with smaller code change