Fix NaN check