From e749474dc3c5a79df4a0301d6b40120178cf967a Mon Sep 17 00:00:00 2001
From: Saul Shanabrook <s.shanabrook@gmail.com>
Date: Sat, 12 Oct 2024 10:55:09 -0400
Subject: [PATCH] Fix query client resolver

---
 package.json       | 2 +-
 src/Visualizer.tsx | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/package.json b/package.json
index 4630e9f..d5d694f 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "egraph-visualizer",
-  "version": "2.0.2",
+  "version": "2.0.3",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/saulshanabrook/egraph-visualizer.git"
diff --git a/src/Visualizer.tsx b/src/Visualizer.tsx
index 12d8fc6..d509b55 100644
--- a/src/Visualizer.tsx
+++ b/src/Visualizer.tsx
@@ -508,7 +508,11 @@ export function render({ model, el }: { el: HTMLElement; model: AnyModel }) {
 export function mount(element: HTMLElement): { render: (egraphs: string[]) => void; unmount: () => void } {
   const root = createRoot(element);
   function render(egraphs: string[]) {
-    root.render(<Visualizer egraphs={egraphs} />);
+    root.render(
+      <QueryClientProvider client={queryClient}>
+        <Visualizer egraphs={egraphs} />
+      </QueryClientProvider>
+    );
   }
 
   function unmount() {