Fixed nan problems