Fixed compile break