Fix import for new column layout