NaN issue fix