Handle NaN values correctly