Fix logic.