Fix for entity generator discriminator column