updated fix for NaN