Fixed compile issues on ubuntu