Created Unassigned: Implementing TPH [26]

I am working with a database that was created using TPH strategy through the designer.

So Company table holds data for Customer class and Lead class using Discriminator field.

I tried doing reverse POCO of this database (I do not have access to the project that created this database) but it just creates a company class. I could modify the generated code and create two classes (Customer and Lead). But then I would have to do that everytime I regenerate the classes.

Is there a better way?

Thanks!