Change class name for $entity