Fix ubuntu build