Fix Ubuntu build