Modified control logic.