ComponentWrapper fix