Improve NaN testing