Fix compilation fails in test file