I would have kept that code and justed intended it in the else clause ...