Fix test compile issue