Fixed compile issue downstream