Fix NaN problem