Fix compile error in test project