Improve UseDef helper to handle FieldRefAssignment