diff --git a/runtime_vis/vis.py b/runtime_vis/vis.py
index 5698385..7321a76 100644
--- a/runtime_vis/vis.py
+++ b/runtime_vis/vis.py
@@ -48,7 +48,7 @@ def profile_batch(self, function: Callable[[int], Any], size: int, dataset: pd.D
 
         row = {'N': size, 'Time': time}
         dataset.loc[len(dataset)] = [size, time]
-        render_plot(dataset)
+        self.render_plot(dataset)
         self.times.append((size,time))
 
         p.disable()
diff --git a/setup.py b/setup.py
index 592e575..50b6a84 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
     name="runtime_vis",
     author="Chris Austin",
     url="https://github.com/chrisxaustin/python-runtime-vis",
-    version="0.8",
+    version="0.9",
     packages=find_packages(),
     install_requires=[
         "seaborn==0.13.2",