Added $entityModel property to collection