fix enterprsie model logic