Fix bad widget wrapping