Fix compile error in test