Fix broken category import