diff --git a/test/useRive.test.tsx b/test/useRive.test.tsx
index 4526c98..ea1f2e0 100644
--- a/test/useRive.test.tsx
+++ b/test/useRive.test.tsx
@@ -2,9 +2,9 @@ import { mocked } from 'jest-mock';
 import { renderHook, act } from '@testing-library/react-hooks';
 
 import useRive from '../src/hooks/useRive';
-import * as rive from '@rive-app/webgl';
+import * as rive from '@rive-app/canvas';
 
-jest.mock('@rive-app/webgl', () => ({
+jest.mock('@rive-app/canvas', () => ({
   Rive: jest.fn().mockImplementation(() => ({
     on: jest.fn(),
     stop: jest.fn(),
diff --git a/test/useStateMachine.test.tsx b/test/useStateMachine.test.tsx
index fcdfe45..8b0fd48 100644
--- a/test/useStateMachine.test.tsx
+++ b/test/useStateMachine.test.tsx
@@ -2,9 +2,9 @@ import { mocked } from 'jest-mock';
 import { renderHook } from '@testing-library/react-hooks';
 
 import useStateMachineInput from '../src/hooks/useStateMachineInput';
-import {Rive, StateMachineInput} from '@rive-app/webgl';
+import {Rive, StateMachineInput} from '@rive-app/canvas';
 
-jest.mock('@rive-app/webgl', () => ({
+jest.mock('@rive-app/canvas', () => ({
   Rive: jest.fn().mockImplementation(() => ({
     on: jest.fn(),
     stop: jest.fn(),