Minor update to utility