Improve NaN handling