fix NaN error for cost