diff --git a/__tests__/cache-restore.test.ts b/__tests__/cache-restore.test.ts
index cb37cc93..7fe60cca 100644
--- a/__tests__/cache-restore.test.ts
+++ b/__tests__/cache-restore.test.ts
@@ -121,7 +121,9 @@ virtualenvs.path = "{cache-dir}/virtualenvs"  # /Users/patrick/Library/Caches/py
             }
           });
 
-          computeKeysSpy.mockResolvedValue('Ubuntu-20.4');
+          computeKeysSpy.mockImplementation(() =>
+            Promise.resolve('Ubuntu-20.4')
+          );
         }
 
         await cacheDistributor.restoreCache();