Fix test compile error