fix some NaN bug