Yup, fixing that logic.