NaN bug fix