fixing NaN error