diff --git a/tests/test_hook/test_empty_cache_hook.py b/tests/test_hook/test_empty_cache_hook.py
index 5d65107022c349c7d5b6b6a6023313af629fbbd7..a09806d0f6697c8f03a4cf60992f3927fa77e80c 100644
--- a/tests/test_hook/test_empty_cache_hook.py
+++ b/tests/test_hook/test_empty_cache_hook.py
@@ -1,5 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-from mock import Mock
+from unittest.mock import Mock
 
 from mmengine.hooks import EmptyCacheHook