fix nan bug