Category import improvement