Fixed compile error in tests