diff --git a/js/CanvasView.js b/js/CanvasView.js
index 3e718e6..ac1c080 100644
--- a/js/CanvasView.js
+++ b/js/CanvasView.js
@@ -852,14 +852,14 @@ async function createCanvasWidget(node, widget, app) {
                     textContent: "Undo",
                     title: "Undo last action",
                     disabled: true,
-                    onclick: () => canvas.canvasState.undo()
+                    onclick: () => canvas.undo()
                 }),
                 $el("button.painter-button", {
                     id: `redo-button-${node.id}`,
                     textContent: "Redo",
                     title: "Redo last undone action",
                     disabled: true,
-                    onclick: () => canvas.canvasState.redo()
+                    onclick: () => canvas.redo()
                 }),
             ]),
             $el("div.painter-separator"),