Update shape capacity when removing ivar and rewriting shape transitions

Since edc7af48acd12666a2945f30901d16b62a39f474, we now no longer have
undef ivar transitions. Instead, we rebuild the shapes table. When we do
this, we need to ensure that we retain our capacities on shapes.

1 files changed, 2 insertions, 1 deletions

diff --git a/shape.c b/shape.c
index 41d3cfcf84..8a7c780e47 100644
--- a/shape.c
+++ b/shape.c
@@ -242,9 +242,10 @@ remove_shape_recursive(VALUE obj, ID id, rb_shape_t * shape, VALUE * removed)
             // has the same attributes as this shape.
             if (new_parent) {
                 rb_shape_t * new_child = get_next_shape_internal(new_parent, shape->edge_name, shape->type);
+                new_child->capacity = shape->capacity;
+
                 if (new_child->type == SHAPE_IVAR) {
                     move_iv(obj, id, shape->next_iv_index - 1, new_child->next_iv_index - 1);
-
                 }
 
                 return new_child;