fixed nan problems