Fixing more than 1 category logic