diff --git a/gallery/example1.pdf b/gallery/example1.pdf
index 96dbf60..22d52c8 100644
Binary files a/gallery/example1.pdf and b/gallery/example1.pdf differ
diff --git a/gallery/example1.typ b/gallery/example1.typ
index bfb9ca0..17e095b 100644
--- a/gallery/example1.typ
+++ b/gallery/example1.typ
@@ -1,6 +1,6 @@
-#import "../src/lib.typ": schema, config
+#import "@preview/rivet:0.3.1": schema, config
 
-#let example = schema.load("/gallery/example1.yaml")
+#let example = schema.load(yaml("./example1.yaml"))
 #schema.render(example, config: config.config(
   full-page: true
 ))
\ No newline at end of file
diff --git a/gallery/example2.pdf b/gallery/example2.pdf
index ee09bba..0692021 100644
Binary files a/gallery/example2.pdf and b/gallery/example2.pdf differ
diff --git a/gallery/example2.typ b/gallery/example2.typ
index 012dd9b..0fddad6 100644
--- a/gallery/example2.typ
+++ b/gallery/example2.typ
@@ -1,6 +1,6 @@
-#import "../src/lib.typ": schema, config
+#import "@preview/rivet:0.3.1": schema, config
 
-#let example = schema.load("/gallery/example2.yaml")
+#let example = schema.load(yaml("./example2.yaml"))
 #schema.render(example, config: config.blueprint(
   full-page: true,
   left-labels: true
diff --git a/gallery/example3.pdf b/gallery/example3.pdf
index 0cb441b..de8082b 100644
Binary files a/gallery/example3.pdf and b/gallery/example3.pdf differ
diff --git a/gallery/example3.typ b/gallery/example3.typ
index cf04b43..c2f28a6 100644
--- a/gallery/example3.typ
+++ b/gallery/example3.typ
@@ -1,6 +1,6 @@
-#import "../src/lib.typ": schema, config
+#import "@preview/rivet:0.3.1": schema, config
 
-#let example = schema.load("/gallery/example1.yaml")
+#let example = schema.load(yaml("/gallery/example1.yaml"))
 //#schema.render(example)
 
 = Chapter 1
diff --git a/gallery/riscv/riscv.typ b/gallery/riscv/riscv.typ
index 0802ad9..7b9f3c5 100644
--- a/gallery/riscv/riscv.typ
+++ b/gallery/riscv/riscv.typ
@@ -1,14 +1,14 @@
-#import "../../src/lib.typ": *
+#import "@preview/rivet:0.3.1": *
 
 #let conf = config.config(
   full-page: true,
   left-labels: true
 )
-#let alu = schema.load("/gallery/riscv/alu_instr.yaml")
+#let alu = schema.load(yaml("./alu_instr.yaml"))
 #schema.render(alu, config: conf)
 
-#let branch = schema.load("/gallery/riscv/branch_instr.yaml")
+#let branch = schema.load(yaml("./branch_instr.yaml"))
 #schema.render(branch, config: conf)
 
-#let mem = schema.load("/gallery/riscv/mem_instr.yaml")
+#let mem = schema.load(yaml("./mem_instr.yaml"))
 #schema.render(mem, config: conf)
diff --git a/gallery/test.pdf b/gallery/test.pdf
index c8d4a11..929dd14 100644
Binary files a/gallery/test.pdf and b/gallery/test.pdf differ
diff --git a/gallery/test.typ b/gallery/test.typ
index 515a846..d5348f8 100644
--- a/gallery/test.typ
+++ b/gallery/test.typ
@@ -1,16 +1,17 @@
-#import "../src/lib.typ": *
+#import "@preview/rivet:0.3.1": *
 
-#let test-yaml = schema.load("/gallery/test.yaml")
+#let test-yaml = schema.load(yaml("./test.yaml"))
 #schema.render(test-yaml, config: config.config(
   full-page: true
 ))
 
-#let test-json = schema.load("/gallery/test.json")
+#let test-json = schema.load(json("./test.json"))
 #schema.render(test-json, config: config.blueprint(
   full-page: true
 ))
 
-#let test-xml = schema.load("/gallery/test.xml")
+#let test-xml-raw = schema.xml-loader.parse(xml("./test.xml").first())
+#let test-xml = schema.load(test-xml-raw)
 #schema.render(test-xml, config: config.dark(
   full-page: true
 ))