fix NaN bug