Fix compile error for test run