fixed nan bug