diff --git a/packages/tron-wallet-snap/src/caching/useCache.test.ts b/packages/tron-wallet-snap/src/caching/useCache.test.ts
index c6ef774c..f01f182c 100644
--- a/packages/tron-wallet-snap/src/caching/useCache.test.ts
+++ b/packages/tron-wallet-snap/src/caching/useCache.test.ts
@@ -174,15 +174,20 @@ describe('useCache', () => {
 
   describe('when the data is cached', () => {
     it('should return the cached result', async () => {
-      // Init the cache with some data
-      jest.spyOn(cache, 'get').mockResolvedValue('test');
+      await withUseCache(
+        async ({ actualExecutionSpy, cache, cachedTestFunction }) => {
+          // Init the cache with some data
+          cache.get.mockResolvedValue('test');
+          // jest.spyOn(cache, 'get').mockResolvedValue('test');
 
-      const result = await cachedTestFunction();
+          const result = await cachedTestFunction();
 
-      expect(result).toBe('test');
-      expect(cache.get).toHaveBeenCalledTimes(1);
-      expect(actualExecutionSpy).not.toHaveBeenCalled();
-      expect(cache.set).not.toHaveBeenCalled();
+          expect(result).toBe('test');
+          expect(cache.get).toHaveBeenCalledTimes(1);
+          expect(actualExecutionSpy).not.toHaveBeenCalled();
+          expect(cache.set).not.toHaveBeenCalled();
+        },
+      );
     });
   });