Could we combine this with the if above and do an `||`?