No, I mean to be able to handle array `$unit`. But nvm, no need, it works with your solution perfectly.