Can you please just make this part of the `else if` chain.