diff --git a/test/Rive.test.tsx b/test/Rive.test.tsx
index 58b863b..9f53b45 100644
--- a/test/Rive.test.tsx
+++ b/test/Rive.test.tsx
@@ -6,6 +6,7 @@ jest.mock('@rive-app/canvas', () => ({
   Rive: jest.fn().mockImplementation(() => ({
     on: jest.fn(),
     stop: jest.fn(),
+    cleanup: jest.fn(),
   })),
   Layout: jest.fn(),
   Fit: {
diff --git a/test/useRive.test.tsx b/test/useRive.test.tsx
index 7c7348b..b81aa26 100644
--- a/test/useRive.test.tsx
+++ b/test/useRive.test.tsx
@@ -38,6 +38,8 @@ describe('useRive', () => {
       stop: jest.fn(),
       stopRendering: jest.fn(),
       startRendering: jest.fn(),
+      cleanup: jest.fn(),
+      resizeToCanvas: jest.fn(),
     };
   });