Yes. Add that logic.