Fix category import