Updates `loc` helper to use new assert syntax