Fix array shapes in vis_mixin.