Fix NaN issue