If I was doing it that way (which would be fine in my opinion) I’d want to do the same for other attacks like the fighter swinging a flametongue sword at whichever layer it is that needs fire damage. I just suggested the attack roll version because it brings it into line with other approaches