Fixing incorrect column assignment