2 files changed, 11 insertions, 9 deletions

diff --git a/src/scene/scene.c b/src/scene/scene.c
index 8c53810..52ddb58 100644
--- a/src/scene/scene.c
+++ b/src/scene/scene.c
@@ -7,19 +7,25 @@
 
 Scene* gfx_make_scene(void) {
   Scene* scene = mem_alloc_scene();
-  scene->root  = gfx_make_node();
+  scene->root  = mem_get_node_index(gfx_make_node());
   return scene;
 }
 
 void gfx_destroy_scene(Scene** scene) {
   assert(scene);
   if (*scene) {
-    gfx_destroy_node(&(*scene)->root);
+    SceneNode* node = mem_get_node((*scene)->root);
+    gfx_destroy_node(&node);
     mem_free_scene(scene);
   }
 }
 
-SceneNode* gfx_get_scene_root(Scene* scene) {
+const SceneNode* gfx_get_scene_root(const Scene* scene) {
   assert(scene);
-  return scene->root;
+  return mem_get_node(scene->root);
+}
+
+SceneNode* gfx_get_scene_root_mut(Scene* scene) {
+  assert(scene);
+  return (SceneNode*)gfx_get_scene_root(scene);
 }
diff --git a/src/scene/scene_impl.h b/src/scene/scene_impl.h
index 06a69f0..157f891 100644
--- a/src/scene/scene_impl.h
+++ b/src/scene/scene_impl.h
@@ -4,10 +4,6 @@
 
 #include "../types.h"
 
-typedef struct SceneNode SceneNode;
-
 typedef struct Scene {
-  SceneNode* root;
-  scene_idx  next;
-  scene_idx  prev;
+  node_idx root;
 } Scene;