Fix NaN bug