Fix the NaN error