Fix nan bugs