could you do it in a `if` clause rather?
i feel this would ass more readability