Fixed compile error in test