fixing issue with EF models

fixing issue with EF models