Fix Ubuntu build error