Makes sense to me, would you like me to add that change to this PR and update the test back to it's original form?