Pull the shape tree out of the vm object

1 files changed, 1 insertions, 1 deletions

diff --git a/yjit.c b/yjit.c
index 431a774880..2d3b7c9e9d 100644
--- a/yjit.c
+++ b/yjit.c
@@ -1105,7 +1105,7 @@ static VALUE
 object_shape_count(rb_execution_context_t *ec, VALUE self)
 {
     // next_shape_id starts from 0, so it's the same as the count
-    return ULONG2NUM((unsigned long)GET_VM()->next_shape_id);
+    return ULONG2NUM((unsigned long)GET_SHAPE_TREE()->next_shape_id);
 }
 
 // Assert that we have the VM lock. Relevant mostly for multi ractor situations.