diff --git a/docs/rules/prefer-called-with.md b/docs/rules/prefer-called-with.md
index 399d511..cc26bf5 100644
--- a/docs/rules/prefer-called-with.md
+++ b/docs/rules/prefer-called-with.md
@@ -14,7 +14,7 @@ Examples of **incorrect** code for this rule:
 
 ```ts
 test('foo', () => {
-  const mock = jest.fn()
+  const mock = vi.fn()
   mock('foo')
   expect(mock).toBeCalled()
   expect(mock).toHaveBeenCalled()
@@ -25,9 +25,9 @@ Examples of **correct** code for this rule:
 
 ```ts
 test('foo', () => {
-  const mock = jest.fn()
+  const mock = vi.fn()
   mock('foo')
   expect(mock).toBeCalledWith('foo')
   expect(mock).toHaveBeenCalledWith('foo')
 })
-```
\ No newline at end of file
+```