Logic fix from previous commit