chore: address a comment for a small refactor proposed

[ata-nas]

This was requested to be added as part of #281, I prefer to address it separately.

Address this comment.