I'm going to rework this logic.