Fix class name in EntityTransformer