Fix to handle NAs.